datasets

Lightning-AI · Mar 15, 2021 · 874055a · 874055a
1 parent 80e2b5d
commit 874055a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pl_examples/__init__.py b/pl_examples/__init__.py
@@ -15,10 +15,10 @@
 _DATASETS_PATH = os.path.join(_PACKAGE_ROOT, 'Datasets')
 
 _TORCHVISION_AVAILABLE = _module_available("torchvision")
-_TORCHVISION_MNIST_AVAILABLE = True
+_TORCHVISION_MNIST_AVAILABLE = _TORCHVISION_AVAILABLE
 _DALI_AVAILABLE = _module_available("nvidia.dali")
 
-if _TORCHVISION_AVAILABLE:
+if _TORCHVISION_MNIST_AVAILABLE:
  try:
  from torchvision.datasets.mnist import MNIST
  MNIST(_DATASETS_PATH, download=True)