[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.configs.train import TrainRLServerPipelineConfig
from lerobot.policies.factory import make_policy from lerobot.policies.factory import make_policy
from lerobot.policies.sac.modeling_sac import SACPolicy from lerobot.policies.sac.modeling_sac import SACPolicy
from lerobot.processor.pipeline import TransitionKey
from lerobot.robots import so100_follower # noqa: F401 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.teleoperators import gamepad, so101_leader # noqa: F401
from lerobot.transport import services_pb2, services_pb2_grpc from lerobot.transport import services_pb2, services_pb2_grpc
from lerobot.transport.utils import ( from lerobot.transport.utils import (
@@ -83,13 +89,6 @@ from lerobot.utils.transition import (
move_state_dict_to_device, move_state_dict_to_device,
move_transition_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 ( from lerobot.utils.utils import (
TimerManager, TimerManager,
get_safe_torch_device, get_safe_torch_device,
+2 -4
View File
@@ -797,13 +797,11 @@ def make_processors(env, cfg):
JointVelocityProcessor(dt=1.0 / cfg.fps), JointVelocityProcessor(dt=1.0 / cfg.fps),
MotorCurrentProcessor(env=env), MotorCurrentProcessor(env=env),
ImageCropResizeProcessor( ImageCropResizeProcessor(
crop_params_dict=cfg.processor.crop_params_dict, crop_params_dict=cfg.processor.crop_params_dict, resize_size=cfg.processor.resize_size
resize_size=cfg.processor.resize_size
), ),
TimeLimitProcessor(max_episode_steps=int(cfg.processor.control_time_s * cfg.fps)), TimeLimitProcessor(max_episode_steps=int(cfg.processor.control_time_s * cfg.fps)),
GripperPenaltyProcessor( GripperPenaltyProcessor(
penalty=cfg.processor.gripper_penalty, penalty=cfg.processor.gripper_penalty, max_gripper_pos=cfg.processor.max_gripper_pos
max_gripper_pos=cfg.processor.max_gripper_pos
), ),
DeviceProcessor(device=cfg.device), DeviceProcessor(device=cfg.device),
] ]