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

Refactor normalization components and update tests

Browse Source 
- Renamed `ObservationNormalizer` to `NormalizerProcessor` and `ActionUnnormalizer` to `UnnormalizerProcessor` for clarity.
- Consolidated normalization logic for both observations and actions into `NormalizerProcessor` and `UnnormalizerProcessor`.
- Updated tests to reflect the new class names and ensure proper functionality of normalization and unnormalization processes.
- Enhanced handling of missing statistics in normalization processes.
This commit is contained in:
Adil Zouitine
2025-07-04 12:09:40 +02:00
parent ed42c71fc3
commit 6830ca7645
3 changed files with 161 additions and 375 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -13,7 +13,7 @@
# 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 .normalize_processor import NormalizationProcessor
from .normalize_processor import NormalizerProcessor, UnnormalizerProcessor
from .observation_processor import (
ImageProcessor,
StateProcessor,
@@ -38,7 +38,8 @@ __all__ = [
"EnvTransition",
"ImageProcessor",
"InfoProcessor",
"NormalizationProcessor",
"NormalizerProcessor",
"UnnormalizerProcessor",
"ObservationProcessor",
"ProcessorStep",
"RenameProcessor",