Update test_rec_system.py

facebookresearch · Apr 25, 2024 · 916af41 · 916af41
1 parent 72ee1ac
commit 916af41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
             )
         )