Update tests/models/test_restore.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Lightning-AI · Apr 19, 2021 · cb021e4 · cb021e4
1 parent 86bb00e
commit cb021e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_restore.py b/tests/models/test_restore.py
@@ -148,7 +148,7 @@ def test_try_resume_from_non_existing_checkpoint(tmpdir):
     trainer.fit(model, datamodule=dm)
     # `True` if resume/restore successfully else `False`
     assert trainer.checkpoint_connector.restore(str(tmpdir / "last.ckpt"), trainer.on_gpu)
-    with pytest.raises(FileNotFoundError):
+    with pytest.raises(FileNotFoundError, match="Aborting training"):
         trainer.checkpoint_connector.restore(str(tmpdir / "last_non_existing.ckpt"), trainer.on_gpu)