From 0db422777c2e15cac08cc8f02175df8205289095 Mon Sep 17 00:00:00 2001 From: Cristobal Eyzaguirre Date: Tue, 22 Oct 2019 20:08:04 -0300 Subject: [PATCH] moved env var to before import (#414) --- pytorch_lightning/logging/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/logging/__init__.py b/pytorch_lightning/logging/__init__.py index 7a265b4c..73960a48 100644 --- a/pytorch_lightning/logging/__init__.py +++ b/pytorch_lightning/logging/__init__.py @@ -10,9 +10,9 @@ try: except ModuleNotFoundError: pass try: - from .comet_logger import CometLogger - # needed to prevent ImportError and duplicated logs. environ["COMET_DISABLE_AUTO_LOGGING"] = "1" + + from .comet_logger import CometLogger except ModuleNotFoundError: - pass + del environ["COMET_DISABLE_AUTO_LOGGING"]