[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-08-02 17:07:19 +00:00
parent cfa672129e
commit ff38a51df9
2 changed files with 24 additions and 27 deletions
+7 -8
View File
@@ -62,8 +62,14 @@ from lerobot.configs import parser
from lerobot.configs.train import TrainRLServerPipelineConfig
from lerobot.policies.factory import make_policy
from lerobot.policies.sac.modeling_sac import SACPolicy
from lerobot.processor.pipeline import TransitionKey
from lerobot.robots import so100_follower # noqa: F401
from lerobot.scripts.rl.gym_manipulator import make_robot_env
from lerobot.scripts.rl.gym_manipulator import (
create_transition,
make_processors,
make_robot_env,
step_env_and_process_transition,
)
from lerobot.teleoperators import gamepad, so101_leader # noqa: F401
from lerobot.transport import services_pb2, services_pb2_grpc
from lerobot.transport.utils import (
@@ -83,13 +89,6 @@ from lerobot.utils.transition import (
move_state_dict_to_device,
move_transition_to_device,
)
from lerobot.processor.pipeline import EnvTransition, TransitionKey
from lerobot.scripts.rl.gym_manipulator import (
create_transition,
make_processors,
step_env_and_process_transition,
)
from lerobot.utils.utils import (
TimerManager,
get_safe_torch_device,
+2 -4
View File
@@ -797,13 +797,11 @@ def make_processors(env, cfg):
JointVelocityProcessor(dt=1.0 / cfg.fps),
MotorCurrentProcessor(env=env),
ImageCropResizeProcessor(
crop_params_dict=cfg.processor.crop_params_dict,
resize_size=cfg.processor.resize_size
crop_params_dict=cfg.processor.crop_params_dict, resize_size=cfg.processor.resize_size
),
TimeLimitProcessor(max_episode_steps=int(cfg.processor.control_time_s * cfg.fps)),
GripperPenaltyProcessor(
penalty=cfg.processor.gripper_penalty,
max_gripper_pos=cfg.processor.max_gripper_pos
penalty=cfg.processor.gripper_penalty, max_gripper_pos=cfg.processor.max_gripper_pos
),
DeviceProcessor(device=cfg.device),
]