From 0ea6aacf8a23169522cbe6ff774bb4b638d6762f Mon Sep 17 00:00:00 2001 From: Pepijn Kooijmans Date: Tue, 7 Apr 2026 11:20:38 +0200 Subject: [PATCH] fix: handle gymnasium < 1.0 without AutoresetMode Made-with: Cursor --- src/lerobot/envs/configs.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/lerobot/envs/configs.py b/src/lerobot/envs/configs.py index 4f18fca6a..1bea2d129 100644 --- a/src/lerobot/envs/configs.py +++ b/src/lerobot/envs/configs.py @@ -22,7 +22,6 @@ from typing import Any import draccus import gymnasium as gym from gymnasium.envs.registration import registry as gym_registry -from gymnasium.vector import AutoresetMode from lerobot.configs.types import FeatureType, PolicyFeature from lerobot.robots import RobotConfig @@ -102,7 +101,11 @@ class EnvConfig(draccus.ChoiceRegistry, abc.ABC): def _make_one(): return gym.make(self.gym_id, disable_env_checker=self.disable_env_checker, **self.gym_kwargs) - vec = env_cls([_make_one for _ in range(n_envs)], autoreset_mode=AutoresetMode.SAME_STEP) + try: + 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}} def get_env_processors(self):