...

Author: Borda

pl_examples/__init__.py

@@ -18,7 +18,7 @@
 _TORCHVISION_MNIST_AVAILABLE = not bool(os.environ.get("PL_USE_MOCKED_MNIST", False))
 _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)

pytorch_lightning/utilities/imports.py

@@ -43,7 +43,8 @@ def _module_available(module_path: str) -> bool:
 
 
 def _compare_version(package: str, op, version) -> bool:
-    """Compare package version with some requirements
+    """
+    Compare package version with some requirements
 
     >>> _compare_version("torch", operator.ge, "0.1")
     True