ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-02 11:51:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(processors): unify import statements by consolidating pipeline imports into the main processor module (#1845)

Browse Source
This commit is contained in:
Steven Palma
2025-09-02 18:26:59 +02:00
committed by GitHub
parent ebb464c255
commit b052843f08
46 changed files with 93 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/lerobot/processor/migrate_policy_normalization.py
View File

@@ -46,11 +46,12 @@ from huggingface_hub import hf_hub_download
from safetensors.torch import load_file as load_safetensors
from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
from lerobot.processor.batch_processor import ToBatchProcessor
from lerobot.processor.device_processor import DeviceProcessor
from lerobot.processor.normalize_processor import NormalizerProcessor, UnnormalizerProcessor
from lerobot.processor.pipeline import RobotProcessor
from lerobot.processor.rename_processor import RenameProcessor
from .batch_processor import ToBatchProcessor
from .device_processor import DeviceProcessor
from .normalize_processor import NormalizerProcessor, UnnormalizerProcessor
from .pipeline import RobotProcessor
from .rename_processor import RenameProcessor
# Policy type to class mapping
POLICY_CLASSES = {