admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-30 07:59:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(pipeline): Move _CFG_NAME along other class member

Browse Source
This commit is contained in:
Adil Zouitine
2025-07-22 10:51:04 +02:00
parent 9fe3a3fb17
commit fb9139b882
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/processor/pipeline.py
+2 -2
View File
@@ -291,6 +291,8 @@ class RobotProcessor(ModelHubMixin):
after_step_hooks: list[Callable[[int, EnvTransition], None]] = field(default_factory=list, repr=False) 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) reset_hooks: list[Callable[[], None]] = field(default_factory=list, repr=False)
_CFG_NAME = "processor.json"
def __call__(self, data: EnvTransition | dict[str, Any]): def __call__(self, data: EnvTransition | dict[str, Any]):
"""Process data through all steps. """Process data through all steps.
@@ -380,8 +382,6 @@ class RobotProcessor(ModelHubMixin):
transition = processor_step(transition) transition = processor_step(transition)
yield self.to_output(transition) if called_with_batch else transition yield self.to_output(transition) if called_with_batch else transition
_CFG_NAME = "processor.json"
def _save_pretrained(self, destination_path: str, **kwargs): def _save_pretrained(self, destination_path: str, **kwargs):
"""Internal save method for ModelHubMixin compatibility.""" """Internal save method for ModelHubMixin compatibility."""
self.save_pretrained(destination_path) self.save_pretrained(destination_path)