Add normalization processor and related components

- Introduced `NormalizationProcessor` to handle both observation normalization and action unnormalization.
- Added `ObservationNormalizer` and `ActionUnnormalizer` classes for specific normalization tasks.
- Updated `__init__.py` to include the new `NormalizationProcessor` in the module exports.
- Enhanced `ObservationProcessor` with registration in the `ProcessorStepRegistry` for better modularity.
- Created `RenameProcessor` for renaming keys in observations, improving flexibility in data processing.
This commit is contained in:
Adil Zouitine
2025-07-03 18:35:14 +02:00
parent ac742c9f0d
commit 8774aec304
4 changed files with 457 additions and 1 deletions

View File

@@ -19,6 +19,7 @@ from .observation_processor import (
StateProcessor,
)
from .pipeline import EnvTransition, ProcessorStep, RobotProcessor
from .normalize_processor import NormalizationProcessor
__all__ = [
"RobotProcessor",
@@ -27,4 +28,5 @@ __all__ = [
"ImageProcessor",
"StateProcessor",
"ObservationProcessor",
"NormalizationProcessor",
]