mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-15 16:49:55 +00:00
chore(pipeline): Move _CFG_NAME along other class member
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user