From 2a7a0e6129b32bd8a89fbed00b69eff4caa98253 Mon Sep 17 00:00:00 2001 From: Adil Zouitine Date: Fri, 4 Jul 2025 11:10:28 +0200 Subject: [PATCH] fix (test): test factory --- tests/envs/test_envs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/envs/test_envs.py b/tests/envs/test_envs.py index 9b07fe64b..9add876c2 100644 --- a/tests/envs/test_envs.py +++ b/tests/envs/test_envs.py @@ -22,7 +22,7 @@ from gymnasium.utils.env_checker import check_env import lerobot 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 tests.utils import require_env @@ -51,7 +51,7 @@ def test_factory(env_name): obs, _ = env.reset() # Process observation using processor - obs_processor = RobotProcessor([ObservationProcessor()]) + obs_processor = RobotProcessor([VanillaObservationProcessor()]) transition = (obs, None, None, None, None, None, None) processed_transition = obs_processor(transition) obs = processed_transition[TransitionIndex.OBSERVATION]