From 9f5ddeb761ba66ad6fbae0ec6080a1bc172e794f Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Thu, 2 Jul 2026 10:58:46 +0200 Subject: [PATCH] fix(style): pre-commit oops --- src/lerobot/policies/factory.py | 2 +- src/lerobot/processor/env_processor.py | 1 + src/lerobot/scripts/lerobot_eval.py | 6 +++--- uv.lock | 22 +++++++++++++--------- 4 files changed, 18 insertions(+), 13 deletions(-) diff --git a/src/lerobot/policies/factory.py b/src/lerobot/policies/factory.py index 9f597469e..1e6558ffe 100644 --- a/src/lerobot/policies/factory.py +++ b/src/lerobot/policies/factory.py @@ -47,9 +47,9 @@ from lerobot.utils.feature_utils import dataset_to_policy_features from .act.configuration_act import ACTConfig from .diffusion.configuration_diffusion import DiffusionConfig from .eo1.configuration_eo1 import EO1Config +from .evo1.configuration_evo1 import Evo1Config from .fastwam.configuration_fastwam import FastWAMConfig from .gaussian_actor.configuration_gaussian_actor import GaussianActorConfig -from .evo1.configuration_evo1 import Evo1Config from .groot.configuration_groot import GrootConfig from .molmoact2.configuration_molmoact2 import MolmoAct2Config from .multi_task_dit.configuration_multi_task_dit import MultiTaskDiTConfig diff --git a/src/lerobot/processor/env_processor.py b/src/lerobot/processor/env_processor.py index 72399e9ab..75cbb79de 100644 --- a/src/lerobot/processor/env_processor.py +++ b/src/lerobot/processor/env_processor.py @@ -78,6 +78,7 @@ class LiberoProcessorStep(ObservationProcessorStep): state = state.float() if state.dim() == 1: state = state.unsqueeze(0) + processed_obs[OBS_STATE] = state return processed_obs diff --git a/src/lerobot/scripts/lerobot_eval.py b/src/lerobot/scripts/lerobot_eval.py index 529def247..b71ea6dac 100644 --- a/src/lerobot/scripts/lerobot_eval.py +++ b/src/lerobot/scripts/lerobot_eval.py @@ -282,9 +282,9 @@ def rollout( action_transition = env_postprocessor(action_transition) action = action_transition[ACTION] - # Convert to CPU / numpy. - action_numpy = _action_to_env_numpy(action) - assert action_numpy.ndim == 2, "Action dimensions should be (batch, action_dim)" + # Convert to CPU / numpy. + action_numpy = _action_to_env_numpy(action) + assert action_numpy.ndim == 2, "Action dimensions should be (batch, action_dim)" # Apply the next action. observation, reward, terminated, truncated, info = env.step(action_numpy) diff --git a/uv.lock b/uv.lock index 7f1741b76..172e7059c 100644 --- a/uv.lock +++ b/uv.lock @@ -2979,14 +2979,14 @@ eo1 = [ evaluation = [ { name = "av" }, ] -fastwam = [ - { name = "diffusers" }, - { name = "transformers" }, -] evo1 = [ { name = "timm" }, { name = "transformers" }, ] +fastwam = [ + { name = "diffusers" }, + { name = "transformers" }, +] feetech = [ { name = "deepdiff" }, { name = "feetech-servo-sdk" }, @@ -3184,6 +3184,9 @@ test = [ { name = "pytest-cov" }, { name = "pytest-timeout" }, ] +timm-dep = [ + { name = "timm" }, +] topreward = [ { name = "transformers" }, ] @@ -3301,8 +3304,8 @@ requires-dist = [ { name = "lerobot", extras = ["diffusers-dep"], marker = "extra == 'vla-jepa'" }, { name = "lerobot", extras = ["diffusion"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["dynamixel"], marker = "extra == 'all'" }, - { name = "lerobot", extras = ["fastwam"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["evo1"], marker = "extra == 'all'" }, + { name = "lerobot", extras = ["fastwam"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["feetech"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["feetech"], marker = "extra == 'hopejr'" }, { name = "lerobot", extras = ["feetech"], marker = "extra == 'lekiwi'" }, @@ -3368,12 +3371,14 @@ requires-dist = [ { name = "lerobot", extras = ["scipy-dep"], marker = "extra == 'wallx'" }, { name = "lerobot", extras = ["smolvla"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["test"], marker = "extra == 'all'" }, + { name = "lerobot", extras = ["timm-dep"], marker = "extra == 'evo1'" }, + { name = "lerobot", extras = ["timm-dep"], marker = "extra == 'groot'" }, { name = "lerobot", extras = ["topreward"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["training"], marker = "extra == 'all'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'annotations'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'eo1'" }, - { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'fastwam'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'evo1'" }, + { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'fastwam'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'groot'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'hilserl'" }, { name = "lerobot", extras = ["transformers-dep"], marker = "extra == 'libero'" }, @@ -3443,8 +3448,7 @@ requires-dist = [ { name = "setuptools", specifier = ">=71.0.0,<81.0.0" }, { name = "teleop", marker = "extra == 'phone'", specifier = ">=0.1.0,<0.2.0" }, { name = "termcolor", specifier = ">=2.4.0,<4.0.0" }, - { name = "timm", marker = "extra == 'evo1'", specifier = ">=1.0.0,<1.1.0" }, - { name = "timm", marker = "extra == 'groot'", specifier = ">=1.0.0,<1.1.0" }, + { name = "timm", marker = "extra == 'timm-dep'", specifier = ">=1.0.0,<1.1.0" }, { name = "torch", marker = "sys_platform != 'linux'", specifier = ">=2.7,<2.12.0" }, { name = "torch", marker = "sys_platform == 'linux'", specifier = ">=2.7,<2.12.0", index = "https://download.pytorch.org/whl/cu128" }, { name = "torchcodec", marker = "(platform_machine == 'arm64' and sys_platform == 'darwin' and extra == 'dataset') or (platform_machine == 'AMD64' and sys_platform == 'linux' and extra == 'dataset') or (platform_machine == 'x86_64' and sys_platform == 'linux' and extra == 'dataset')", specifier = ">=0.3.0,<0.12.0" }, @@ -3457,7 +3461,7 @@ requires-dist = [ { name = "transformers", marker = "extra == 'transformers-dep'", specifier = ">=5.4.0,<5.6.0" }, { name = "wandb", marker = "extra == 'training'", specifier = ">=0.24.0,<0.28.0" }, ] -provides-extras = ["dataset", "training", "hardware", "viz", "core-scripts", "evaluation", "dataset-viz", "av-dep", "pygame-dep", "placo-dep", "transformers-dep", "grpcio-dep", "accelerate-dep", "can-dep", "peft-dep", "scipy-dep", "diffusers-dep", "qwen-vl-utils-dep", "matplotlib-dep", "pyserial-dep", "deepdiff-dep", "pynput-dep", "pyzmq-dep", "motorbridge-dep", "motorbridge-smart-servo-dep", "feetech", "dynamixel", "damiao", "robstride", "openarms", "gamepad", "hopejr", "lekiwi", "unitree-g1", "reachy2", "rebot", "kinematics", "intelrealsense", "phone", "diffusion", "wallx", "pi", "molmoact2", "smolvla", "multi-task-dit", "groot", "sarm", "robometer", "topreward", "xvla", "eo1", "fastwam", "evo1", "hilserl", "vla-jepa", "async", "peft", "annotations", "dev", "notebook", "test", "video-benchmark", "aloha", "pusht", "libero", "metaworld", "all"] +provides-extras = ["dataset", "training", "hardware", "viz", "core-scripts", "evaluation", "dataset-viz", "av-dep", "pygame-dep", "placo-dep", "transformers-dep", "grpcio-dep", "accelerate-dep", "can-dep", "peft-dep", "scipy-dep", "diffusers-dep", "qwen-vl-utils-dep", "matplotlib-dep", "pyserial-dep", "deepdiff-dep", "pynput-dep", "pyzmq-dep", "motorbridge-dep", "motorbridge-smart-servo-dep", "timm-dep", "feetech", "dynamixel", "damiao", "robstride", "openarms", "gamepad", "hopejr", "lekiwi", "unitree-g1", "reachy2", "rebot", "kinematics", "intelrealsense", "phone", "diffusion", "wallx", "pi", "molmoact2", "smolvla", "multi-task-dit", "groot", "sarm", "robometer", "topreward", "xvla", "eo1", "fastwam", "evo1", "hilserl", "vla-jepa", "async", "peft", "annotations", "dev", "notebook", "test", "video-benchmark", "aloha", "pusht", "libero", "metaworld", "all"] [[package]] name = "librt"