admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 00:29:52 +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
tests/processor/test_device_processor.py
+2 -3
View File
@@ -19,8 +19,7 @@ import pytest
import torch
from lerobot.configs.types import FeatureType, PolicyFeature
from lerobot.processor import DeviceProcessor, RobotProcessor
from lerobot.processor.pipeline import TransitionKey
from lerobot.processor import DeviceProcessor, RobotProcessor, TransitionKey
def create_transition(
@@ -358,7 +357,7 @@ def test_save_and_load_pretrained():
def test_registry_functionality():
"""Test that DeviceProcessor is properly registered."""
from lerobot.processor.pipeline import ProcessorStepRegistry
from lerobot.processor import ProcessorStepRegistry
# Check that DeviceProcessor is registered
registered_class = ProcessorStepRegistry.get("device_processor")