From 916af411d88ff96e1cb3d05223c7070963b5612d Mon Sep 17 00:00:00 2001 From: dkorenkevych Date: Thu, 25 Apr 2024 09:55:32 -0700 Subject: [PATCH] Update test_rec_system.py --- test/unit/test_tutorials/test_rec_system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, ) )