diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py index d1e87332ccc71..386b8f1e23ea9 100644 --- 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)