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"]