update

Author: justusschock

tests/tests_pytorch/callbacks/progress/test_tqdm_progress_bar.py

@@ -519,6 +519,7 @@ def test_tqdm_progress_bar_print_no_train(tqdm_write, tmp_path):
         limit_predict_batches=1,
         max_steps=1,
         callbacks=[bar],
+        devices=1,
     )
 
     trainer.validate(model)

tests/tests_pytorch/loggers/conftest.py

@@ -125,6 +125,9 @@ def __getitem__(self, item):
         def __setitem__(self, key, value):
             pass
 
+        def wait(self):
+            pass
+
     run_mock = MagicMock(spec=RunType, exists=Mock(return_value=False), wait=Mock(), get_structure=MagicMock())
     run_mock.get_root_object.return_value = run_mock