Update test_rec_system.py

facebookresearch · Apr 25, 2024 · e58eec8 · e58eec8
1 parent daf049b
commit e58eec8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/unit/test_tutorials/test_rec_system.py b/test/unit/test_tutorials/test_rec_system.py
@@ -67,7 +67,7 @@
 This environment's underlying model was pre-trained using the MIND dataset (Wu et al. 2020).
 The model is defined by class `SequenceClassificationModel` below.
 The model's state dict is saved in
-tutorials/single_item_recommender_system_example/env_model_state_dict.pt
+tutorials/single_item_recommender_system_example/env_model_state_dict
 
 Each data point is:
 - A history of impressions clicked by a user
@@ -195,7 +195,7 @@ def test_rec_system(self) -> None:
         if os.path.exists("../Pearl"):
             # Github CI tests
             print("Pearl directory exists")
-            model_dir = "Pearl/tutorials/single_item_recommender_system_example/"
+            model_dir = "tutorials/single_item_recommender_system_example/"
         else:
             # Meta internal tests
             print("pearl directory exists")