...

Lightning-AI · Mar 22, 2021 · 24cea3e · 24cea3e
1 parent 156ea8c
commit 24cea3e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pl_examples/__init__.py b/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)

diff --git a/pytorch_lightning/utilities/imports.py b/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