chore(processor): rename specialized processor -> XYZProcessorStep (#1852)

This commit is contained in:
Steven Palma
2025-09-03 17:30:47 +02:00
committed by GitHub
parent 8c796b39f5
commit d893bf1e30
20 changed files with 89 additions and 85 deletions

View File

@@ -23,10 +23,10 @@ from lerobot.configs.types import FeatureType, PolicyFeature
from lerobot.constants import ACTION, OBS_STATE
from lerobot.model.kinematics import RobotKinematics
from lerobot.processor import (
ActionProcessor,
ComplementaryDataProcessor,
ActionProcessorStep,
ComplementaryDataProcessorStep,
EnvTransition,
ObservationProcessor,
ObservationProcessorStep,
ProcessorStep,
ProcessorStepRegistry,
TransitionKey,
@@ -36,7 +36,7 @@ from lerobot.robots.robot import Robot
@ProcessorStepRegistry.register("ee_reference_and_delta")
@dataclass
class EEReferenceAndDelta(ActionProcessor):
class EEReferenceAndDelta(ActionProcessorStep):
"""
Compute the desired end-effector pose from the target pose and the current pose.
@@ -159,7 +159,7 @@ class EEReferenceAndDelta(ActionProcessor):
@ProcessorStepRegistry.register("ee_bounds_and_safety")
@dataclass
class EEBoundsAndSafety(ActionProcessor):
class EEBoundsAndSafety(ActionProcessorStep):
"""
Clip the end-effector pose to the bounds and check for jumps.
@@ -389,7 +389,7 @@ class GripperVelocityToJoint(ProcessorStep):
@ProcessorStepRegistry.register("forward_kinematics_joints_to_ee")
@dataclass
class ForwardKinematicsJointsToEE(ObservationProcessor):
class ForwardKinematicsJointsToEE(ObservationProcessorStep):
"""
Compute the end-effector pose from the joint positions.
@@ -433,7 +433,7 @@ class ForwardKinematicsJointsToEE(ObservationProcessor):
@ProcessorStepRegistry.register("add_robot_observation")
@dataclass
class AddRobotObservationAsComplimentaryData(ComplementaryDataProcessor):
class AddRobotObservationAsComplimentaryData(ComplementaryDataProcessorStep):
"""
Read the robot's current observation and insert it into the transition as complementary data.