Disable Gym passive checker by default in stable_worldmodel env creation

2026-04-09 10:18:54 +00:00
parent 96e17a13af
commit 20ffb3492b
1 changed files with 7 additions and 1 deletions
--- a/.venv/lib/python3.10/site-packages/stable_worldmodel/world.py
+++ b/.venv/lib/python3.10/site-packages/stable_worldmodel/world.py
@@ -48,7 +48,13 @@ def _make_env(env_name, max_episode_steps, wrappers, **kwargs):
        >>> wrappers = [partial(MegaWrapper, image_shape=(64, 64))]
        >>> env = _make_env("CartPole-v1", max_episode_steps=500, wrappers=wrappers)
    """
-    env = gym.make(env_name, max_episode_steps=max_episode_steps, **kwargs)
+    disable_env_checker = kwargs.pop('disable_env_checker', True)
+    env = gym.make(
+        env_name,
+        max_episode_steps=max_episode_steps,
+        disable_env_checker=disable_env_checker,
+        **kwargs,
+    )
    for wrapper in wrappers:
        env = wrapper(env)
    return env