Test fixes

Author: carmocca

tests/tests_pytorch/core/test_datamodules.py

@@ -230,7 +230,7 @@ def load_state_dict(self, state_dict: Dict[str, Any]) -> None:
 
     for trainer_fn in TrainerFn:
         trainer.state.fn = trainer_fn
-        trainer._restore_modules_and_callbacks(checkpoint_path)
+        trainer._checkpoint_connector._restore_modules_and_callbacks(checkpoint_path)
         assert dm.my_state_dict == {"my": "state_dict"}
 
 

tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py

@@ -52,7 +52,8 @@ def test_preloaded_checkpoint_lifecycle(tmpdir):
     assert isinstance(connector._loaded_checkpoint, dict)
     trainer.state.fn = TrainerFn.FITTING
     connector.resume_end()
-    assert not connector._ckpt_path
+    # not cleared until next restoration, as the user might access it through `trainer.ckpt_path`
+    assert connector._ckpt_path == ckpt_path
     assert not connector._loaded_checkpoint