From 15be98655877912f145797feff41f36a823aa00d Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Wed, 9 Jun 2021 15:07:02 +0200 Subject: [PATCH] add logger to __all__ (#6854) --- tests/loggers/test_all.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py index a18b180be1..1bad12b1f9 100644 --- a/tests/loggers/test_all.py +++ b/tests/loggers/test_all.py @@ -24,6 +24,7 @@ import tests.helpers.utils as tutils from pytorch_lightning import Callback, Trainer from pytorch_lightning.loggers import ( CometLogger, + CSVLogger, MLFlowLogger, NeptuneLogger, TensorBoardLogger, @@ -233,6 +234,7 @@ def _test_loggers_save_dir_and_weights_save_path(tmpdir, logger_class): "logger_class", [ CometLogger, + CSVLogger, MLFlowLogger, NeptuneLogger, TensorBoardLogger, @@ -325,6 +327,7 @@ class RankZeroLoggerCheck(Callback): @pytest.mark.parametrize( "logger_class", [ CometLogger, + CSVLogger, MLFlowLogger, NeptuneLogger, TensorBoardLogger,