chore(processor): rename specialized processor -> XYZProcessorStep (#1852)

This commit is contained in:
Steven Palma
2025-09-03 17:30:47 +02:00
committed by GitHub
parent 8c796b39f5
commit d893bf1e30
20 changed files with 89 additions and 85 deletions
+2 -2
View File
@@ -27,7 +27,7 @@ from lerobot.policies.sac.reward_model.processor_classifier import make_classifi
from lerobot.processor import (
DataProcessorPipeline,
DeviceProcessor,
IdentityProcessor,
IdentityProcessorStep,
NormalizerProcessor,
TransitionKey,
)
@@ -94,7 +94,7 @@ def test_make_classifier_processor_basic():
# Check steps in postprocessor
assert len(postprocessor.steps) == 2
assert isinstance(postprocessor.steps[0], DeviceProcessor)
assert isinstance(postprocessor.steps[1], IdentityProcessor)
assert isinstance(postprocessor.steps[1], IdentityProcessorStep)
def test_classifier_processor_normalization():
+4 -4
View File
@@ -22,7 +22,7 @@ import torch
from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
from lerobot.processor import (
DataProcessorPipeline,
IdentityProcessor,
IdentityProcessorStep,
NormalizerProcessor,
TransitionKey,
UnnormalizerProcessor,
@@ -1008,7 +1008,7 @@ def test_hotswap_stats_basic_functionality():
# Create processors
normalizer = NormalizerProcessor(features=features, norm_map=norm_map, stats=initial_stats)
unnormalizer = UnnormalizerProcessor(features=features, norm_map=norm_map, stats=initial_stats)
identity = IdentityProcessor()
identity = IdentityProcessorStep()
# Create robot processor
robot_processor = DataProcessorPipeline(steps=[normalizer, unnormalizer, identity])
@@ -1089,7 +1089,7 @@ def test_hotswap_stats_only_affects_normalizer_steps():
# Create mixed steps
normalizer = NormalizerProcessor(features=features, norm_map=norm_map, stats=stats)
unnormalizer = UnnormalizerProcessor(features=features, norm_map=norm_map, stats=stats)
identity = IdentityProcessor()
identity = IdentityProcessorStep()
robot_processor = DataProcessorPipeline(steps=[normalizer, identity, unnormalizer])
@@ -1135,7 +1135,7 @@ def test_hotswap_stats_no_normalizer_steps():
}
# Create processor with only identity steps
robot_processor = DataProcessorPipeline(steps=[IdentityProcessor(), IdentityProcessor()])
robot_processor = DataProcessorPipeline(steps=[IdentityProcessorStep(), IdentityProcessorStep()])
# Hotswap stats - should work without error
new_processor = hotswap_stats(robot_processor, stats)