...

Lightning-AI · Mar 14, 2021 · 5965162 · 5965162
1 parent e0de6e5
commit 5965162
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_lightning/utilities/imports.py b/pytorch_lightning/utilities/imports.py
@@ -48,6 +48,8 @@ def _compare_version(package: str, op, version) -> bool:
     >>> _compare_version("torch", operator.ge, "0.1")
     True
     """
+    if not _module_available(package):
+        return False
     try:
         pkg = importlib.import_module(package)
         assert hasattr(pkg, '__version__')