mirror of
https://github.com/huggingface/lerobot.git
synced 2026-06-03 04:11:24 +00:00
refactor(processors): add transform_features method to various processors (#1843)
This commit is contained in:
@@ -45,6 +45,9 @@ class MapTensorToDeltaActionDict(ActionProcessor):
|
||||
delta_action["action.gripper"] = action[3]
|
||||
return delta_action
|
||||
|
||||
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
|
||||
return features
|
||||
|
||||
|
||||
@ProcessorStepRegistry.register("map_delta_action_to_robot_action")
|
||||
@dataclass
|
||||
|
||||
Reference in New Issue
Block a user