ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 04:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

5
src/lerobot/processor/pipeline.py
View File

@@ -169,7 +169,7 @@ class ProcessorStep(ABC):
def reset(self) -> None:
return None
# TODO(Steven): Consider making this abstract so it is more explicit
@abstractmethod
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
return features
@@ -1091,3 +1091,6 @@ class IdentityProcessor(ProcessorStep):
def __call__(self, transition: EnvTransition) -> EnvTransition:
return transition
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
return features