refactor(processors): Standardize processor naming conventions

- Updated processor names across various files to use a consistent "robot_preprocessor" and "robot_postprocessor" format.
- Modified the make_processor functions in factory, act, diffusion, pi0, pi0fast, sac, smolvla, tdmpc, and vqbet to reflect the new naming scheme.
- Enhanced the pipeline configuration to align with the updated processor names, improving clarity and maintainability.
This commit is contained in:
Adil Zouitine
2025-08-05 14:40:19 +02:00
committed by Steven Palma
parent 5326ffe77e
commit 87890cbf38
10 changed files with 21 additions and 22 deletions
+2 -2
View File
@@ -45,6 +45,6 @@ def make_act_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="act_preprocessor"), RobotProcessor(
steps=output_steps, name="act_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
@@ -46,6 +46,6 @@ def make_diffusion_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="diffusion_preprocessor"), RobotProcessor(
steps=output_steps, name="diffusion_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
+2 -2
View File
@@ -143,12 +143,12 @@ def make_processor(
return (
RobotProcessor.from_pretrained(
source=pretrained_path,
config_filename=kwargs.get("preprocessor_config_filename", "preprocessor.json"),
config_filename=kwargs.get("preprocessor_config_filename", "robot_preprocessor.json"),
overrides=kwargs.get("preprocessor_overrides", {}),
),
RobotProcessor.from_pretrained(
source=pretrained_path,
config_filename=kwargs.get("postprocessor_config_filename", "postprocessor.json"),
config_filename=kwargs.get("postprocessor_config_filename", "robot_postprocessor.json"),
overrides=kwargs.get("postprocessor_overrides", {}),
),
)
+2 -2
View File
@@ -115,6 +115,6 @@ def make_pi0_processor(
),
]
return RobotProcessor(steps=input_steps, name="pi0_preprocessor"), RobotProcessor(
steps=output_steps, name="pi0_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
@@ -46,6 +46,6 @@ def make_pi0_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="pi0_preprocessor"), RobotProcessor(
steps=output_steps, name="pi0_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
+2 -2
View File
@@ -47,6 +47,6 @@ def make_sac_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="sac_preprocessor"), RobotProcessor(
steps=output_steps, name="sac_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
@@ -57,8 +57,8 @@ def make_smolvla_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="smolvla_preprocessor"), RobotProcessor(
steps=output_steps, name="smolvla_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
@@ -46,6 +46,6 @@ def make_tdmpc_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="tdmpc_preprocessor"), RobotProcessor(
steps=output_steps, name="tdmpc_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
@@ -47,6 +47,6 @@ def make_vqbet_processor(
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
),
]
return RobotProcessor(steps=input_steps, name="vqbet_preprocessor"), RobotProcessor(
steps=output_steps, name="vqbet_postprocessor"
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
steps=output_steps, name="robot_postprocessor"
)
+3 -4
View File
@@ -587,10 +587,9 @@ class RobotProcessor(ModelHubMixin):
if config_filename is None:
# Try common config names
common_names = [
"processor.json",
"preprocessor.json",
"postprocessor.json",
"robotprocessor.json",
"robot_processor.json",
"robot_preprocessor.json",
"robot_postprocessor.json",
]
config_path = None
for name in common_names: