Update pytorch_lightning/loggers/base.py

Co-authored-by: Jirka Borovec <[email protected]>
Lightning-AI · Jun 17, 2020 · 5cd682c · 5cd682c
1 parent 93baf9a
commit 5cd682c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pytorch_lightning/loggers/base.py b/pytorch_lightning/loggers/base.py
@@ -9,7 +9,10 @@
 import torch
 
 from pytorch_lightning.utilities import rank_zero_only
-from omegaconf import DictConfig
+try:
+    from omegaconf import Container
+except ImportError:
+    Container = None
 
 
 class LightningLoggerBase(ABC):