Skip to content

Commit 07b97f4

Browse files
update names in scripts
1 parent 278ec93 commit 07b97f4

File tree

3 files changed

+7
-119
lines changed

3 files changed

+7
-119
lines changed

experiments/generic/run_generic_agent.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ def main():
2323
parser.add_argument("--n-relaunch", type=int, default=3)
2424
parser.add_argument("--parallel-backend", type=str, default="ray")
2525
parser.add_argument("--reproducibility-mode", action="store_true")
26+
parser.add_argument("--debug", action="store_true")
2627

2728
args = parser.parse_args()
2829

@@ -31,11 +32,12 @@ def main():
3132
benchmark = DEFAULT_BENCHMARKS[args.benchmark]()
3233

3334
##################### Shuffle env args list, pick subset
34-
import numpy as np
35-
rng = np.random.default_rng(42)
36-
rng.shuffle(benchmark.env_args_list)
37-
benchmark.env_args_list = benchmark.env_args_list[:33]
38-
#####################
35+
if args.debug:
36+
import numpy as np
37+
rng = np.random.default_rng(42)
38+
rng.shuffle(benchmark.env_args_list)
39+
benchmark.env_args_list = benchmark.env_args_list[:33]
40+
#####################
3941

4042
# for env_args in benchmark.env_args_list:
4143
# env_args.max_steps = 100

experiments/hinter/run_hinter_agent.py

Lines changed: 0 additions & 84 deletions
This file was deleted.

experiments/hinter/run_hinter_agent.sh

Lines changed: 0 additions & 30 deletions
This file was deleted.

0 commit comments

Comments
 (0)