From 3e7c9d7afce1c13c5bf4fd56ccf0e35c56344144 Mon Sep 17 00:00:00 2001 From: Adil Zouitine Date: Fri, 1 Aug 2025 09:18:16 +0200 Subject: [PATCH] feat(batch_processor): Add feature_contract method to ToBatchProcessor - Introduced feature_contract method that returns features without modification, maintaining the no-op behavior of the processor. - This addition enhances the flexibility of the ToBatchProcessor for future feature processing needs. --- src/lerobot/processor/batch_processor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/lerobot/processor/batch_processor.py b/src/lerobot/processor/batch_processor.py index 2c496fe8b..1f356e9bf 100644 --- a/src/lerobot/processor/batch_processor.py +++ b/src/lerobot/processor/batch_processor.py @@ -121,3 +121,7 @@ class ToBatchProcessor: def reset(self) -> None: """Reset processor state (no-op for this processor).""" pass + + def feature_contract(self, features: dict[str, Any]) -> dict[str, Any]: + """Return features (no-op for this processor).""" + return features