tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update factory.py

Browse Source
This commit is contained in:
Michel Aractingi
2024-12-31 11:38:53 +01:00
committed by GitHub
parent f3450aea41
commit 245e844d83
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/envs/factory.py
View File

@@ -79,7 +79,7 @@ def make_maniskill_env(cfg: DictConfig, n_envs: int | None = None) -> gym.vector
) )
# cfg.env_cfg.control_mode = cfg.eval_env_cfg.control_mode = env.control_mode # cfg.env_cfg.control_mode = cfg.eval_env_cfg.control_mode = env.control_mode
env = ManiSkillVectorEnv(env, ignore_terminations=True) env = ManiSkillVectorEnv(env, ignore_terminations=True)
env = PixelWrapper(cfg, env, n_envs) # env = PixelWrapper(cfg, env, n_envs)
env._max_episode_steps = env.max_episode_steps = 50 # gym_utils.find_max_episode_steps_value(env) env._max_episode_steps = env.max_episode_steps = 50 # gym_utils.find_max_episode_steps_value(env)
env.unwrapped.metadata["render_fps"] = 20 env.unwrapped.metadata["render_fps"] = 20