diff --git a/pufferlib/environments/cogames/environment.py b/pufferlib/environments/cogames/environment.py index 0fbe47595..75a8bae76 100644 --- a/pufferlib/environments/cogames/environment.py +++ b/pufferlib/environments/cogames/environment.py @@ -17,11 +17,9 @@ def make(name="cogames.cogs_v_clips.machina_1.open_world", variants=None, cogs=N 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() simulator.add_event_handler(StatsTracker(NoopStatsWriter())) env = PufferMettaGridEnv(simulator=simulator, cfg=env_cfg, buf=buf, seed=seed or 0) - env.render_mode = render if seed: env.reset(seed) return env