Make test_tube optional (Lightning-AI#274)

lezwon · Oct 1, 2019 · 8a24722 · 8a24722
1 parent 324c28e
commit 8a24722
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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: