admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 19:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use direct AutoresetMode import for gymnasium compat

Browse Source 
Made-with: Cursor
This commit is contained in:
Pepijn Kooijmans
2026-04-07 11:19:17 +02:00
committed by Pepijn
parent 1fad71cee8
commit d8e0eaaf42
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/envs/configs.py
+2 -6
View File
@@ -22,6 +22,7 @@ from typing import Any
import draccus import draccus
import gymnasium as gym import gymnasium as gym
from gymnasium.envs.registration import registry as gym_registry from gymnasium.envs.registration import registry as gym_registry
from gymnasium.vector import AutoresetMode
from lerobot.configs.types import FeatureType, PolicyFeature from lerobot.configs.types import FeatureType, PolicyFeature
from lerobot.robots import RobotConfig from lerobot.robots import RobotConfig
@@ -101,12 +102,7 @@ class EnvConfig(draccus.ChoiceRegistry, abc.ABC):
def _make_one(): def _make_one():
return gym.make(self.gym_id, disable_env_checker=self.disable_env_checker, **self.gym_kwargs) return gym.make(self.gym_id, disable_env_checker=self.disable_env_checker, **self.gym_kwargs)
try: vec = env_cls([_make_one for _ in range(n_envs)], autoreset_mode=AutoresetMode.SAME_STEP)
from gymnasium.vector import AutoresetMode
vec = env_cls([_make_one for _ in range(n_envs)], autoreset_mode=AutoresetMode.SAME_STEP)
except ImportError:
vec = env_cls([_make_one for _ in range(n_envs)])
return {self.type: {0: vec}} return {self.type: {0: vec}}
def get_env_processors(self): def get_env_processors(self):