mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 19:19:56 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
@@ -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),
|
||||||
]
|
]
|
||||||
|
|||||||
Reference in New Issue
Block a user