From b04061f80c662fac67743b77aa502cae506509f7 Mon Sep 17 00:00:00 2001 From: dkorenkevych Date: Fri, 26 Apr 2024 09:30:59 -0700 Subject: [PATCH 1/2] Update run_cb_benchmarks.py --- pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py b/pearl/utils/scripts/cb_benchmark/run_cb_benchmarks.py index 24319fe2..df36f767 100644 --- 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, ) From 203783216a85c02ae5f1ca7151e3de352f2b341e Mon Sep 17 00:00:00 2001 From: dkorenkevych Date: Fri, 26 Apr 2024 09:31:24 -0700 Subject: [PATCH 2/2] Update cb_benchmark_config.py --- pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py b/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py index db394466..11a9925c 100644 --- a/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py +++ b/pearl/utils/scripts/cb_benchmark/cb_benchmark_config.py @@ -28,7 +28,7 @@ from pearl.policy_learners.exploration_modules.contextual_bandits.ucb_exploration import ( UCBExploration, ) -from pearl.test.utils import prefix_dir +from test.utils import prefix_dir from pearl.utils.instantiations.environments.contextual_bandit_uci_environment import ( SLCBEnvironment, )