From a804755e6e018c8fecf4be1a1bb2be6604e82b7a Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Thu, 23 Jan 2020 16:36:40 +0000 Subject: [PATCH] update logger init (#727) * update logger init * formatting --- pytorch_lightning/logging/__init__.py | 16 +++++++--------- pytorch_lightning/root_module/__init__.py | 3 ++- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/pytorch_lightning/logging/__init__.py b/pytorch_lightning/logging/__init__.py index 2697e4e6..e9fe3eef 100644 --- a/pytorch_lightning/logging/__init__.py +++ b/pytorch_lightning/logging/__init__.py @@ -76,41 +76,39 @@ from os import environ from .base import LightningLoggerBase, rank_zero_only from .tensorboard import TensorBoardLogger -all = [] +loggers = ['TensorBoardLogger'] try: # needed to prevent ImportError and duplicated logs. environ["COMET_DISABLE_AUTO_LOGGING"] = "1" from .comet import CometLogger - all.append('CometLogger') + loggers.append('CometLogger') except ImportError: del environ["COMET_DISABLE_AUTO_LOGGING"] try: from .mlflow import MLFlowLogger - all.append('MLFlowLogger') + loggers.append('MLFlowLogger') except ImportError: pass try: from .neptune import NeptuneLogger - all.append('NeptuneLogger') + loggers.append('NeptuneLogger') except ImportError: pass -all.append('TensorBoardLogger') - try: from .test_tube import TestTubeLogger - all.append('TestTubeLogger') + loggers.append('TestTubeLogger') except ImportError: pass try: from .wandb import WandbLogger - all.append('WandbLogger') + loggers.append('WandbLogger') except ImportError: pass -__all__ = all +__all__ = loggers diff --git a/pytorch_lightning/root_module/__init__.py b/pytorch_lightning/root_module/__init__.py index 1d0f5f06..2ab9f1a0 100644 --- a/pytorch_lightning/root_module/__init__.py +++ b/pytorch_lightning/root_module/__init__.py @@ -8,4 +8,5 @@ warnings.warn("`root_module` package has been renamed to `core` since v0.6.0" " and will be removed in v0.8.0", DeprecationWarning) from pytorch_lightning.core import ( # noqa: E402 - decorators, grads, hooks, root_module, memory, model_saving) + decorators, grads, hooks, root_module, memory, model_saving +)