From 4df4d4cc03a687ae44ebe8d80669824a867b6d75 Mon Sep 17 00:00:00 2001 From: Nic Eggert Date: Wed, 30 Oct 2019 11:03:52 -0500 Subject: [PATCH] Catch exceptions when optional dependencies are missing (#442) --- pytorch_lightning/logging/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/logging/__init__.py b/pytorch_lightning/logging/__init__.py index 73960a48..dc47188a 100644 --- a/pytorch_lightning/logging/__init__.py +++ b/pytorch_lightning/logging/__init__.py @@ -3,16 +3,16 @@ from .base import LightningLoggerBase, rank_zero_only try: from .test_tube_logger import TestTubeLogger -except ModuleNotFoundError: +except ImportError: pass try: from .mlflow_logger import MLFlowLogger -except ModuleNotFoundError: +except ImportError: pass try: # needed to prevent ImportError and duplicated logs. environ["COMET_DISABLE_AUTO_LOGGING"] = "1" from .comet_logger import CometLogger -except ModuleNotFoundError: +except ImportError: del environ["COMET_DISABLE_AUTO_LOGGING"]