simplify

Author: relh

pufferlib/config/cogames.ini

@@ -1,6 +1,6 @@
 [base]
 package = cogames
-env_name = cogames.cogs-v-clips.machina_1.open_world
+env_name = machina_1.open_world
 policy_name = Policy
 rnn_name = Recurrent
 

pufferlib/environments/cogames/environment.py

@@ -8,18 +8,12 @@
 from mettagrid.util.stats_writer import NoopStatsWriter
 
 
-def env_creator(name="cogames.cogs-v-clips"):
+def env_creator(name="machina_1.open_world"):
     return functools.partial(make, name=name)
 
 
-def make(name="cogames.cogs-v-clips.machina_1.open_world", variants=None, cogs=None, render_mode="auto", seed=None, buf=None):
-    # Strip package prefixes
-    parts = name.split(".")
-    while parts and parts[0].replace("-", "_") in {"cogames", "cogs_v_clips"}:
-        parts.pop(0)
-    mission_name = ".".join(parts) if parts else "training_facility.harvest"
-
-    _, env_cfg, _ = get_mission(mission_name, variants_arg=variants, cogs=cogs)
+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)
 
     render = "none" if render_mode == "auto" else "unicode" if render_mode in {"human", "ansi"} else render_mode
     simulator = Simulator()