refactor(processors): add transform_features method to various processors (#1843)

This commit is contained in:
Steven Palma
2025-09-02 17:15:01 +02:00
committed by GitHub
parent 645c87e3a9
commit 2914ae2a96
11 changed files with 71 additions and 2 deletions
@@ -220,6 +220,9 @@ class EEBoundsAndSafety(ActionProcessor):
self._last_pos = None
self._last_twist = None
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
return features
@ProcessorStepRegistry.register("inverse_kinematics_ee_to_joints")
@dataclass
@@ -444,3 +447,6 @@ class AddRobotObservationAsComplimentaryData(ComplementaryDataProcessor):
if isinstance(k, str) and k.endswith(".pos")
}
return new_comp
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
return features