chore(pipeline): Move _CFG_NAME along other class member

This commit is contained in:
Adil Zouitine
2025-07-22 10:51:04 +02:00
parent 9fe3a3fb17
commit fb9139b882
+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)
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)