ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-01 03:11:29 +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

5
src/lerobot/processor/device_processor.py
View File

@@ -19,10 +19,11 @@ from typing import Any
import torch
from lerobot.configs.types import PolicyFeature
from lerobot.processor.core import EnvTransition, TransitionKey
from lerobot.processor.pipeline import ProcessorStep, ProcessorStepRegistry
from lerobot.utils.utils import get_safe_torch_device
from .core import EnvTransition, TransitionKey
from .pipeline import ProcessorStep, ProcessorStepRegistry
@ProcessorStepRegistry.register("device_processor")
@dataclass