mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-19 10:40:04 +00:00
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:
committed by
Steven Palma
parent
5326ffe77e
commit
87890cbf38
@@ -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"
|
||||
)
|
||||
|
||||
@@ -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", {}),
|
||||
),
|
||||
)
|
||||
|
||||
@@ -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"
|
||||
)
|
||||
|
||||
@@ -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"
|
||||
)
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user