Update cb_benchmark_config.py

facebookresearch · Apr 25, 2024 · 5eab3b9 · 5eab3b9
1 parent e236e74
commit 5eab3b9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py b/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py
@@ -31,8 +31,10 @@
 from pearl.utils.instantiations.environments.contextual_bandit_uci_environment import (
     SLCBEnvironment,
 )
-
-DATA_PATH: str = "./utils/instantiations/environments/uci_datasets"
+if os.path.exists("../Pearl"):
+    DATA_PATH: str = "pearl/utils/instantiations/environments/uci_datasets"
+else:
+    DATA_PATH: str = "./utils/instantiations/environments/uci_datasets"
 
 """
 Experiment config