checking if runners work

ucl-dark · Oct 24, 2023 · 86c8345 · 86c8345
1 parent 64cbf18
commit 86c8345
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pax/experiment.py b/pax/experiment.py
@@ -307,7 +307,7 @@ def runner_setup(args, env, agents, save_dir, logger):
         return IPDITMEvalRunner(agents, env, save_dir, args)
 
     if args.runner in ["evo", "evo_mixed_lr", "evo_hardstop", "evo_mixed_payoff", "evo_mixed_ipd_payoff",
-    "evo_mixed_payoff_gen", "evo_mixed_payoff_input", "evo_mixed_payoff_pred", "evo_scanned", "evo_mixed_payoff_only_opp", "multishaper_evo"]:
+    "evo_mixed_payoff_gen", "evo_mixed_payoff_input", "evo_scanned", "evo_mixed_payoff_only_opp", "multishaper_evo"]:
         agent1, _ = agents
         algo = args.es.algo
         strategies = {"CMA_ES", "OpenES", "PGPE", "SimpleGA"}

diff --git a/pax/runners/runner_evo_scanned.py b/pax/runners/runner_evo_scanned.py
@@ -37,6 +37,8 @@ class EvoScannedRunner:
     It composes together agents, watchers, and the environment.
     Within the init, we declare vmaps and pmaps for training.
     The environment provided must conform to a meta-environment.
+    This runner also scans over the evolutionary step, which leads to longer compilation time,
+    shorter run time and logging is not possible.
     Args:
         agents (Tuple[agents]):
             The set of agents that will run in the experiment. Note, ordering is