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

View File

@@ -18,6 +18,7 @@ from typing import Any
import torch
from lerobot.configs.types import PolicyFeature
from lerobot.processor.core import EnvTransition, TransitionKey
from lerobot.processor.pipeline import ProcessorStep, ProcessorStepRegistry
from lerobot.utils.utils import get_safe_torch_device
@@ -127,3 +128,6 @@ class DeviceProcessor(ProcessorStep):
def get_config(self) -> dict[str, Any]:
"""Return configuration for serialization."""
return {"device": self.device, "float_dtype": self.float_dtype}
def transform_features(self, features: dict[str, PolicyFeature]) -> dict[str, PolicyFeature]:
return features