Update test_rec_system.py

facebookresearch · Apr 25, 2024 · daf049b · daf049b
1 parent 6b70386
commit daf049b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/unit/test_tutorials/test_rec_system.py b/test/unit/test_tutorials/test_rec_system.py
@@ -192,14 +192,14 @@ def test_rec_system(self) -> None:
         model = SequenceClassificationModel(100).to(device)
         print("current dir", os.getcwd())
         print("dirs list", os.listdir())
-        if os.path.exists("pearl"):
-            # Meta internal tests
-            print("pearl directory exists")
-            model_dir = "pearl/tutorials/single_item_recommender_system_example/"
-        else:
+        if os.path.exists("../Pearl"):
             # Github CI tests
             print("Pearl directory exists")
             model_dir = "Pearl/tutorials/single_item_recommender_system_example/"
+        else:
+            # Meta internal tests
+            print("pearl directory exists")
+            model_dir = "pearl/tutorials/single_item_recommender_system_example/"
         model.load_state_dict(
             # Note: in the tutorial the directory "pearl" must be replaced by "Pearl"
             torch.load(