From fb9139b8821b9284d69a1f4cf7285762f3698914 Mon Sep 17 00:00:00 2001 From: Adil Zouitine Date: Tue, 22 Jul 2025 10:51:04 +0200 Subject: [PATCH] chore(pipeline): Move _CFG_NAME along other class member --- src/lerobot/processor/pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)