qihuanye/lewm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Disable Gym passive checker by default in stable_worldmodel env creation

Browse Source
This commit is contained in:
qihuanye
2026-04-09 10:18:54 +00:00
parent 96e17a13af
commit 20ffb3492b
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.venv/lib/python3.10/site-packages/stable_worldmodel/world.py
View File

@@ -48,7 +48,13 @@ def _make_env(env_name, max_episode_steps, wrappers, **kwargs):
>>> wrappers = [partial(MegaWrapper, image_shape=(64, 64))] >>> wrappers = [partial(MegaWrapper, image_shape=(64, 64))]
>>> env = _make_env("CartPole-v1", max_episode_steps=500, wrappers=wrappers) >>> 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: for wrapper in wrappers:
env = wrapper(env) env = wrapper(env)
return env return env