Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add WandB and Neptune to exports from loggers module (#18157) 

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 <aedu.waelchli@gmail.com>
This commit is contained in:
Jonathan Rayner 2023-07-27 12:38:30 +01:00 committed by GitHub
parent dfb687aa28
commit 1585264437
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lightning/pytorch/loggers/__init__.py
View File

@ -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.csv_logs import CSVLogger
from lightning.pytorch.loggers.logger import Logger from lightning.pytorch.loggers.logger import Logger
from lightning.pytorch.loggers.mlflow import _MLFLOW_AVAILABLE, MLFlowLogger # noqa: F401 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.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: if _COMET_AVAILABLE:
__all__.append("CometLogger") __all__.append("CometLogger")