tests: disable lightning logging in unit tests (#38)

tilman151 · Jul 4, 2023 · c3aa8ed · c3aa8ed
1 parent aedc4d8
commit c3aa8ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_approach/utils.py b/tests/test_approach/utils.py
@@ -9,7 +9,7 @@ def checkpoint(approach, ckpt_path, max_rul=50):
     dm = rul_datasets.RulDataModule(
         rul_datasets.reader.DummyReader(1, max_rul=max_rul), 32
     )
-    trainer = pl.Trainer(max_epochs=0, num_sanity_val_steps=0)
+    trainer = pl.Trainer(max_epochs=0, num_sanity_val_steps=0, logger=False)
     trainer.fit(approach, dm)
     trainer.save_checkpoint(ckpt_path)