From f11fc9da916de9f4134d06bcdca6645b7dade912 Mon Sep 17 00:00:00 2001 From: Jake Grigsby Date: Tue, 8 Oct 2024 21:24:28 +0000 Subject: [PATCH] fix version control issue in examples --- examples/05_kshot_metaworld.py | 2 +- examples/09_ale.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/examples/05_kshot_metaworld.py b/examples/05_kshot_metaworld.py index 799210e..df4edfb 100644 --- a/examples/05_kshot_metaworld.py +++ b/examples/05_kshot_metaworld.py @@ -32,7 +32,7 @@ def add_cli(parser): args = parser.parse_args() config = { - "amago.envs.env_utils.ExplorationWrapper.steps_anneal": 2_000_000, + "BilevelEpsilonGreedy.steps_anneal": 2_000_000, "amago.nets.tstep_encoders.FFTstepEncoder.hide_rl2s": args.hide_rl2s, "amago.agent.Agent.reward_multiplier": 1.0, # delete the next three lines to use the paper settings, which were diff --git a/examples/09_ale.py b/examples/09_ale.py index dd0e80f..42407e3 100644 --- a/examples/09_ale.py +++ b/examples/09_ale.py @@ -79,9 +79,7 @@ def make_atari_game(game_name): game_name = games[actor % len(games)] env_funcs.append(partial(make_atari_game, game_name)) - group_name = ( - f"{args.run_name}_{','.join(games)}_atari_l_{args.max_seq_len}_cnn_{args.cnn}" - ) + group_name = f"{args.run_name}_atari_l_{args.max_seq_len}_cnn_{args.cnn}" for trial in range(args.trials): run_name = group_name + f"_trial_{trial}" experiment = create_experiment_from_cli(