variant parsing

Author: relh

pufferlib/config/cogames.ini

@@ -1,6 +1,6 @@
 [base]
 package = cogames
-env_name = machina_1.open_world
+env_name = cogames.cogs_v_clips.training_facility.harvest cogames.cogs_v_clips.training_facility.assemble cogames.cogs_v_clips.machina_1.open_world
 policy_name = Policy
 rnn_name = Recurrent
 

pufferlib/environments/cogames/environment.py

@@ -8,12 +8,14 @@
 from mettagrid.util.stats_writer import NoopStatsWriter
 
 
-def env_creator(name="machina_1.open_world"):
+def env_creator(name="cogames.cogs_v_clips.machina_1.open_world"):
     return functools.partial(make, name=name)
 
 
-def make(name="machina_1.open_world", variants=None, cogs=None, render_mode="auto", seed=None, buf=None):
-    _, env_cfg, _ = get_mission(name, variants_arg=variants, cogs=cogs)
+def make(name="cogames.cogs_v_clips.machina_1.open_world", variants=None, cogs=None, render_mode="auto", seed=None, buf=None):
+    mission_name = name.removeprefix("cogames.cogs_v_clips.") if name.startswith("cogames.cogs_v_clips.") else name
+    variants = variants.split() if isinstance(variants, str) else variants
+    _, env_cfg, _ = get_mission(mission_name, variants_arg=variants, cogs=cogs)
 
     render = "none" if render_mode == "auto" else "unicode" if render_mode in {"human", "ansi"} else render_mode
     simulator = Simulator()