try fix the test

Lightning-AI · Nov 23, 2020 · 6b08489 · 6b08489
1 parent 8a3d576
commit 6b08489
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -369,4 +369,4 @@ def test_logger_with_prefix_all(tmpdir, monkeypatch):
     with mock.patch('pytorch_lightning.loggers.wandb.wandb') as wandb:
         logger = _instantiate_logger(WandbLogger, save_idr=tmpdir, prefix=prefix)
         logger.log_metrics({"test": 1.0}, step=0)
-        wandb.init().log.assert_called_once_with({'tmp-test': 1.0}, step=0)
+        logger.experiment.log.assert_called_once_with({'tmp-test': 1.0}, step=0)