moved COMET_DISABLE_AUTO_LOGGING out of modeule for flake8 compliance (#410)

* moved COMET_DISABLE_AUTO_LOGGING out of modeule for flake8 compliance

* Update __init__.py

pytorch_lightning/logging/__init__.py

@@ -1,3 +1,4 @@
+from os import environ
 from .base import LightningLoggerBase, rank_zero_only
 
 try:
@@ -10,5 +11,8 @@ except ModuleNotFoundError:
     pass
 try:
     from .comet_logger import CometLogger
+
+    # needed to prevent ImportError and duplicated logs.
+    environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
 except ModuleNotFoundError:
     pass

pytorch_lightning/logging/comet_logger.py

@@ -4,9 +4,6 @@ from comet_ml import Experiment as CometExperiment
 
 from .base import LightningLoggerBase, rank_zero_only
 
-# needed to prevent ImportError and duplicated logs.
-environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
-
 
 class CometLogger(LightningLoggerBase):
     def __init__(self, *args, **kwargs):