diff --git a/src/lerobot/processor/pipeline.py b/src/lerobot/processor/pipeline.py index 3bb005841..888ec3b36 100644 --- a/src/lerobot/processor/pipeline.py +++ b/src/lerobot/processor/pipeline.py @@ -291,6 +291,8 @@ class RobotProcessor(ModelHubMixin): after_step_hooks: list[Callable[[int, EnvTransition], None]] = field(default_factory=list, repr=False) reset_hooks: list[Callable[[], None]] = field(default_factory=list, repr=False) + _CFG_NAME = "processor.json" + def __call__(self, data: EnvTransition | dict[str, Any]): """Process data through all steps. @@ -380,8 +382,6 @@ class RobotProcessor(ModelHubMixin): transition = processor_step(transition) yield self.to_output(transition) if called_with_batch else transition - _CFG_NAME = "processor.json" - def _save_pretrained(self, destination_path: str, **kwargs): """Internal save method for ModelHubMixin compatibility.""" self.save_pretrained(destination_path)