Update run_cb_benchmarks.py

facebookresearch · Apr 25, 2024 · e236e74 · e236e74
1 parent 7036f96
commit e236e74
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py b/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py
@@ -277,15 +277,22 @@ def run_cb_benchmarks(
     uci_data_path = "./utils/instantiations/environments/uci_datasets"
     print("current_dir", os.getcwd())
     print("dirs list", os.listdir())
-    print("dir exists", os.path.exists("./utils/instantiations/environments/"))
+
+    if os.path.exists("../Pearl"):
+        uci_data_path = "pearl/utils/instantiations/environments/uci_datasets"
+        save_results_path: str = "pearl/utils/scripts/cb_benchmark/experiments_results"
+        print("dir exists", os.path.exists("pearl/utils/instantiations/environments/"))
+    else:
+        uci_data_path = "utils/instantiations/environments/uci_datasets"
+        save_results_path: str = "utils/scripts/cb_benchmark/experiments_results"
     if not os.path.exists(uci_data_path):
         os.makedirs(uci_data_path)
 
     # Download UCI data
     download_uci_data(data_path=uci_data_path)
 
     # Create folder for result if it does not already exist
-    save_results_path: str = "./utils/scripts/cb_benchmark/experiments_results"
+
     if not os.path.exists(save_results_path):
         os.makedirs(save_results_path)