fix version control issue in examples

UT-Austin-RPL · Oct 8, 2024 · f11fc9d · f11fc9d
1 parent 15a6f76
commit f11fc9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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(