admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): import issue

Browse Source
This commit is contained in:
Adil Zouitine
2025-07-04 11:25:07 +02:00
parent e0139065bd
commit ed42c71fc3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/eval.py
+2 -2
View File
@@ -72,7 +72,7 @@ from lerobot.envs.utils import add_envs_task, check_env_attributes_and_types
from lerobot.policies.factory import make_policy from lerobot.policies.factory import make_policy
from lerobot.policies.pretrained import PreTrainedPolicy from lerobot.policies.pretrained import PreTrainedPolicy
from lerobot.policies.utils import get_device_from_parameters from lerobot.policies.utils import get_device_from_parameters
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 lerobot.utils.io_utils import write_video from lerobot.utils.io_utils import write_video
from lerobot.utils.random_utils import set_seed from lerobot.utils.random_utils import set_seed
@@ -131,7 +131,7 @@ def rollout(
render_callback(env) render_callback(env)
# Create observation processing processor # Create observation processing processor
obs_processor = RobotProcessor([ObservationProcessor()]) obs_processor = RobotProcessor([VanillaObservationProcessor()])
all_observations = [] all_observations = []
all_actions = [] all_actions = []