Files
pytorch-lightning/pytorch_lightning/logging/__init__.py
T
Cristobal Eyzaguirre 2924ebeda5 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
2019-10-22 13:06:07 -04:00

19 lines
445 B
Python

from os import environ
from .base import LightningLoggerBase, rank_zero_only
try:
from .test_tube_logger import TestTubeLogger
except ModuleNotFoundError:
pass
try:
from .mlflow_logger import MLFlowLogger
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