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")