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

add logger to __all__ (#6854)

This commit is contained in:
Jirka Borovec 2021-06-09 15:07:02 +02:00 committed by GitHub
parent 6fee9262ff
commit 15be986558
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/loggers/test_all.py
View File

@ -24,6 +24,7 @@ import tests.helpers.utils as tutils
from pytorch_lightning import Callback, Trainer from pytorch_lightning import Callback, Trainer
from pytorch_lightning.loggers import ( from pytorch_lightning.loggers import (
CometLogger, CometLogger,
CSVLogger,
MLFlowLogger, MLFlowLogger,
NeptuneLogger, NeptuneLogger,
TensorBoardLogger, TensorBoardLogger,
@ -233,6 +234,7 @@ def _test_loggers_save_dir_and_weights_save_path(tmpdir, logger_class):
"logger_class", "logger_class",
[ [
CometLogger, CometLogger,
CSVLogger,
MLFlowLogger, MLFlowLogger,
NeptuneLogger, NeptuneLogger,
TensorBoardLogger, TensorBoardLogger,
@ -325,6 +327,7 @@ class RankZeroLoggerCheck(Callback):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"logger_class", [ "logger_class", [
CometLogger, CometLogger,
CSVLogger,
MLFlowLogger, MLFlowLogger,
NeptuneLogger, NeptuneLogger,
TensorBoardLogger, TensorBoardLogger,