Update tests/models/test_restore.py

Co-authored-by: Carlos Mocholí <[email protected]>

Author: tchaton

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)