add logger to __all__ (#6854)

Lightning-AI · Jun 9, 2021 · 15be986 · 15be986
1 parent 6fee926
commit 15be986
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -24,6 +24,7 @@
 from pytorch_lightning import Callback, Trainer
 from pytorch_lightning.loggers import (
     CometLogger,
+    CSVLogger,
     MLFlowLogger,
     NeptuneLogger,
     TensorBoardLogger,
@@ -233,6 +234,7 @@ def name(self):
     "logger_class",
     [
         CometLogger,
+        CSVLogger,
         MLFlowLogger,
         NeptuneLogger,
         TensorBoardLogger,
@@ -325,6 +327,7 @@ def on_train_batch_start(self, trainer, pl_module, batch, batch_idx, dataloader_
 @pytest.mark.parametrize(
     "logger_class", [
         CometLogger,
+        CSVLogger,
         MLFlowLogger,
         NeptuneLogger,
         TensorBoardLogger,