diff --git a/test/unit/test_tutorials/test_rec_system.py b/test/unit/test_tutorials/test_rec_system.py index 5ebf351c..7c044ab2 100644 --- a/test/unit/test_tutorials/test_rec_system.py +++ b/test/unit/test_tutorials/test_rec_system.py @@ -202,7 +202,7 @@ def test_rec_system(self) -> None: model.load_state_dict( # Note: in the tutorial the directory "pearl" must be replaced by "Pearl" torch.load( - os.path.join("model_dir", "env_model_state_dict.pt"), + os.path.join(model_dir, "env_model_state_dict.pt"), weights_only=True, ) )