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

refactor(pipeline): minor improvements (#1684)

Browse Source 
* chore(pipeline): remove unused features + device torch + envtransition keys

* refactor(pipeline): ImageProcessor & StateProcessor are both implemented directly in VanillaObservationPRocessor

* refactor(pipeline): RenameProcessor now inherits from ObservationProcessor + remove unused code

* test(pipeline): fix broken test after refactors

* docs(pipeline): update docstrings VanillaObservationProcessor

* chore(pipeline): move None check to base pipeline classes
This commit is contained in:
Steven Palma
2025-08-06 14:00:13 +02:00
committed by GitHub
parent 7beb040e8e
commit fd4ae3466b
8 changed files with 165 additions and 421 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/lerobot/processor/__init__.py
View File

@@ -16,11 +16,7 @@
from .device_processor import DeviceProcessor
from .normalize_processor import NormalizerProcessor, UnnormalizerProcessor
from .observation_processor import (
ImageProcessor,
StateProcessor,
VanillaObservationProcessor,
)
from .observation_processor import VanillaObservationProcessor
from .pipeline import (
ActionProcessor,
DoneProcessor,
@@ -43,7 +39,6 @@ __all__ = [
"DoneProcessor",
"EnvTransition",
"IdentityProcessor",
"ImageProcessor",
"InfoProcessor",
"NormalizerProcessor",
"UnnormalizerProcessor",
@@ -53,7 +48,6 @@ __all__ = [
"RenameProcessor",
"RewardProcessor",
"RobotProcessor",
"StateProcessor",
"TransitionKey",
"TruncatedProcessor",
"VanillaObservationProcessor",