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

fix (test): test factory

Browse Source
This commit is contained in:
Adil Zouitine
2025-07-04 11:10:28 +02:00
parent 9f33791b19
commit 2a7a0e6129
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/envs/test_envs.py
+2 -2
View File
@@ -22,7 +22,7 @@ from gymnasium.utils.env_checker import check_env
import lerobot import lerobot
from lerobot.envs.factory import make_env, make_env_config from lerobot.envs.factory import make_env, make_env_config
from lerobot.processor.observation_processor import ObservationProcessor from lerobot.processor.observation_processor import VanillaObservationProcessor
from lerobot.processor.pipeline import RobotProcessor, TransitionIndex from lerobot.processor.pipeline import RobotProcessor, TransitionIndex
from tests.utils import require_env from tests.utils import require_env
@@ -51,7 +51,7 @@ def test_factory(env_name):
obs, _ = env.reset() obs, _ = env.reset()
# Process observation using processor # Process observation using processor
obs_processor = RobotProcessor([ObservationProcessor()]) obs_processor = RobotProcessor([VanillaObservationProcessor()])
transition = (obs, None, None, None, None, None, None) transition = (obs, None, None, None, None, None, None)
processed_transition = obs_processor(transition) processed_transition = obs_processor(transition)
obs = processed_transition[TransitionIndex.OBSERVATION] obs = processed_transition[TransitionIndex.OBSERVATION]