Update run_cb_benchmarks.py

facebookresearch · Apr 26, 2024 · b04061f · b04061f
1 parent 9a8ea26
commit b04061f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py b/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py
@@ -29,7 +29,7 @@
 from pearl.replay_buffers.contextual_bandits.discrete_contextual_bandit_replay_buffer import (
     DiscreteContextualBanditReplayBuffer,
 )
-from pearl.test.utils import prefix_dir
+from test.utils import prefix_dir
 from pearl.utils.instantiations.environments.contextual_bandit_uci_environment import (
     SLCBEnvironment,
 )