From 158526443706f22d332e63c069878664e9450eeb Mon Sep 17 00:00:00 2001 From: Jonathan Rayner Date: Thu, 27 Jul 2023 12:38:30 +0100 Subject: [PATCH] Add WandB and Neptune to exports from loggers module (#18157) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Adrian Wälchli --- src/lightning/pytorch/loggers/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lightning/pytorch/loggers/__init__.py b/src/lightning/pytorch/loggers/__init__.py index 4d275660ee..0c3c454b73 100644 --- a/src/lightning/pytorch/loggers/__init__.py +++ b/src/lightning/pytorch/loggers/__init__.py @@ -17,11 +17,11 @@ from lightning.pytorch.loggers.comet import _COMET_AVAILABLE, CometLogger # noq from lightning.pytorch.loggers.csv_logs import CSVLogger from lightning.pytorch.loggers.logger import Logger from lightning.pytorch.loggers.mlflow import _MLFLOW_AVAILABLE, MLFlowLogger # noqa: F401 -from lightning.pytorch.loggers.neptune import NeptuneLogger # noqa: F401 +from lightning.pytorch.loggers.neptune import NeptuneLogger from lightning.pytorch.loggers.tensorboard import TensorBoardLogger -from lightning.pytorch.loggers.wandb import WandbLogger # noqa: F401 +from lightning.pytorch.loggers.wandb import WandbLogger -__all__ = ["CSVLogger", "Logger", "TensorBoardLogger"] +__all__ = ["CSVLogger", "Logger", "TensorBoardLogger", "WandbLogger", "NeptuneLogger"] if _COMET_AVAILABLE: __all__.append("CometLogger")