mirror of
https://github.com/huggingface/lerobot.git
synced 2026-06-02 11:51:25 +00:00
f2b79656eb
- Updated the EnvTransition structure to use a dictionary format instead of a tuple, enhancing readability and maintainability. - Replaced instances of TransitionIndex with TransitionKey for accessing transition components. - Adjusted related processing functions and tests to accommodate the new dictionary format, ensuring consistent handling of transitions across the codebase.
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Copyright 2025 The HuggingFace Inc. team. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from .device_processor import DeviceProcessor
|
|
from .normalize_processor import NormalizerProcessor, UnnormalizerProcessor
|
|
from .observation_processor import (
|
|
ImageProcessor,
|
|
StateProcessor,
|
|
VanillaObservationProcessor,
|
|
)
|
|
from .pipeline import (
|
|
ActionProcessor,
|
|
DoneProcessor,
|
|
EnvTransition,
|
|
IdentityProcessor,
|
|
InfoProcessor,
|
|
ObservationProcessor,
|
|
ProcessorStep,
|
|
ProcessorStepRegistry,
|
|
RewardProcessor,
|
|
RobotProcessor,
|
|
TransitionKey,
|
|
TruncatedProcessor,
|
|
)
|
|
from .rename_processor import RenameProcessor
|
|
|
|
__all__ = [
|
|
"ActionProcessor",
|
|
"DeviceProcessor",
|
|
"DoneProcessor",
|
|
"EnvTransition",
|
|
"IdentityProcessor",
|
|
"ImageProcessor",
|
|
"InfoProcessor",
|
|
"NormalizerProcessor",
|
|
"UnnormalizerProcessor",
|
|
"ObservationProcessor",
|
|
"ProcessorStep",
|
|
"ProcessorStepRegistry",
|
|
"RenameProcessor",
|
|
"RewardProcessor",
|
|
"RobotProcessor",
|
|
"StateProcessor",
|
|
"TransitionKey",
|
|
"TruncatedProcessor",
|
|
"VanillaObservationProcessor",
|
|
]
|