mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 19:19:56 +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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="act_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="act_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="diffusion_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="diffusion_postprocessor"
|
steps=output_steps, name="robot_postprocessor"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -143,12 +143,12 @@ def make_processor(
|
|||||||
return (
|
return (
|
||||||
RobotProcessor.from_pretrained(
|
RobotProcessor.from_pretrained(
|
||||||
source=pretrained_path,
|
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", {}),
|
overrides=kwargs.get("preprocessor_overrides", {}),
|
||||||
),
|
),
|
||||||
RobotProcessor.from_pretrained(
|
RobotProcessor.from_pretrained(
|
||||||
source=pretrained_path,
|
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", {}),
|
overrides=kwargs.get("postprocessor_overrides", {}),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -115,6 +115,6 @@ def make_pi0_processor(
|
|||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
return RobotProcessor(steps=input_steps, name="pi0_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="pi0_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="pi0_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="pi0_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="sac_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="sac_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="smolvla_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="smolvla_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="tdmpc_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="tdmpc_postprocessor"
|
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
|
features=config.output_features, norm_map=config.normalization_mapping, stats=dataset_stats
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
return RobotProcessor(steps=input_steps, name="vqbet_preprocessor"), RobotProcessor(
|
return RobotProcessor(steps=input_steps, name="robot_preprocessor"), RobotProcessor(
|
||||||
steps=output_steps, name="vqbet_postprocessor"
|
steps=output_steps, name="robot_postprocessor"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -587,10 +587,9 @@ class RobotProcessor(ModelHubMixin):
|
|||||||
if config_filename is None:
|
if config_filename is None:
|
||||||
# Try common config names
|
# Try common config names
|
||||||
common_names = [
|
common_names = [
|
||||||
"processor.json",
|
"robot_processor.json",
|
||||||
"preprocessor.json",
|
"robot_preprocessor.json",
|
||||||
"postprocessor.json",
|
"robot_postprocessor.json",
|
||||||
"robotprocessor.json",
|
|
||||||
]
|
]
|
||||||
config_path = None
|
config_path = None
|
||||||
for name in common_names:
|
for name in common_names:
|
||||||
|
|||||||
Reference in New Issue
Block a user