From 5cd682c557eb9e966e33d4f4fe991833f6ed7206 Mon Sep 17 00:00:00 2001
From: Siavash Sakhavi <s.sakhavi@gmail.com>
Date: Wed, 17 Jun 2020 16:25:04 +0800
Subject: [PATCH] Update pytorch_lightning/loggers/base.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
---
 pytorch_lightning/loggers/base.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pytorch_lightning/loggers/base.py b/pytorch_lightning/loggers/base.py
index 16eb65a55c099e..626f074212080f 100644
--- 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):