refactor(normalization): Clean up imports in normalize_processor.py

This commit is contained in:
Adil Zouitine
2025-08-01 08:56:20 +02:00
committed by Steven Palma
parent 8ff95be04c
commit e9cb779eab
+1 -2
View File
@@ -1,6 +1,5 @@
from __future__ import annotations
from collections.abc import Mapping
from copy import deepcopy
from dataclasses import dataclass, field
@@ -12,7 +11,7 @@ from torch import Tensor
from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
from lerobot.datasets.lerobot_dataset import LeRobotDataset
from lerobot.processor.pipeline import EnvTransition, ProcessorStepRegistry, TransitionKey, RobotProcessor
from lerobot.processor.pipeline import EnvTransition, ProcessorStepRegistry, RobotProcessor, TransitionKey
def _convert_stats_to_tensors(stats: dict[str, dict[str, Any]]) -> dict[str, dict[str, Tensor]]: