diff --git a/pytorch_lightning/logging/__init__.py b/pytorch_lightning/logging/__init__.py index 89e3d435..b8e2fcff 100644 --- a/pytorch_lightning/logging/__init__.py +++ b/pytorch_lightning/logging/__init__.py @@ -1,6 +1,9 @@ from .base import LightningLoggerBase, rank_zero_only -from .test_tube_logger import TestTubeLogger +try: + from .test_tube_logger import TestTubeLogger +except ModuleNotFoundError: + pass try: from .mlflow_logger import MLFlowLogger except ModuleNotFoundError: