From 9dde8829e68cec49d2ce08b2fcedeec4353ad58c Mon Sep 17 00:00:00 2001
From: Michel Aractingi <michel.aractingi@huggingface.co>
Date: Wed, 2 Jul 2025 17:10:56 +0200
Subject: [PATCH] style nit

---
 src/lerobot/datasets/utils.py       | 2 +-
 tests/datasets/test_datasets.py     | 4 ++--
 tests/fixtures/dataset_factories.py | 2 +-
 tests/policies/test_policies.py     | 6 +++---
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/lerobot/datasets/utils.py b/src/lerobot/datasets/utils.py
index 6223b98a2..bd4005443 100644
--- a/src/lerobot/datasets/utils.py
+++ b/src/lerobot/datasets/utils.py
@@ -40,6 +40,7 @@ from huggingface_hub.errors import RevisionNotFoundError
 from PIL import Image as PILImage
 from torchvision import transforms
 
+from lerobot.configs.types import FeatureType, PolicyFeature
 from lerobot.datasets.backward_compatibility import (
     V21_MESSAGE,
     BackwardCompatibilityError,
@@ -47,7 +48,6 @@ from lerobot.datasets.backward_compatibility import (
 )
 from lerobot.robots import Robot
 from lerobot.utils.utils import is_valid_numpy_dtype_string
-from lerobot.configs.types import FeatureType, PolicyFeature
 
 DEFAULT_CHUNK_SIZE = 1000  # Max number of files per chunk
 DEFAULT_DATA_FILE_SIZE_IN_MB = 100  # Max size per file
diff --git a/tests/datasets/test_datasets.py b/tests/datasets/test_datasets.py
index 905cb23cf..fe0a1b290 100644
--- a/tests/datasets/test_datasets.py
+++ b/tests/datasets/test_datasets.py
@@ -26,6 +26,8 @@ from PIL import Image
 from safetensors.torch import load_file
 
 import lerobot
+from lerobot.configs.default import DatasetConfig
+from lerobot.configs.train import TrainPipelineConfig
 from lerobot.datasets.factory import make_dataset
 from lerobot.datasets.image_writer import image_array_to_pil_image
 from lerobot.datasets.lerobot_dataset import (
@@ -39,8 +41,6 @@ from lerobot.datasets.utils import (
 from lerobot.envs.factory import make_env_config
 from lerobot.policies.factory import make_policy_config
 from lerobot.robots import make_robot_from_config
-from lerobot.configs.default import DatasetConfig
-from lerobot.configs.train import TrainPipelineConfig
 from tests.fixtures.constants import DUMMY_CHW, DUMMY_HWC, DUMMY_REPO_ID
 from tests.mocks.mock_robot import MockRobotConfig
 from tests.utils import require_x86_64_kernel
diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py
index ac79139fb..2459b0bf6 100644
--- a/tests/fixtures/dataset_factories.py
+++ b/tests/fixtures/dataset_factories.py
@@ -26,6 +26,7 @@ import pytest
 import torch
 from datasets import Dataset
 
+from lerobot.common.datasets.video_utils import encode_video_frames
 from lerobot.datasets.lerobot_dataset import CODEBASE_VERSION, LeRobotDataset, LeRobotDatasetMetadata
 from lerobot.datasets.utils import (
     DEFAULT_CHUNK_SIZE,
@@ -38,7 +39,6 @@ from lerobot.datasets.utils import (
     get_hf_features_from_features,
     hf_transform_to_torch,
 )
-from lerobot.common.datasets.video_utils import encode_video_frames
 from tests.fixtures.constants import (
     DEFAULT_FPS,
     DUMMY_CAMERA_FEATURES,
diff --git a/tests/policies/test_policies.py b/tests/policies/test_policies.py
index 57ac0edc1..3a8c6a224 100644
--- a/tests/policies/test_policies.py
+++ b/tests/policies/test_policies.py
@@ -23,6 +23,9 @@ import torch
 from safetensors.torch import load_file
 
 from lerobot import available_policies
+from lerobot.configs.default import DatasetConfig
+from lerobot.configs.train import TrainPipelineConfig
+from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
 from lerobot.datasets.factory import make_dataset
 from lerobot.datasets.utils import cycle, dataset_to_policy_features
 from lerobot.envs.factory import make_env, make_env_config
@@ -37,9 +40,6 @@ from lerobot.policies.factory import (
 from lerobot.policies.normalize import Normalize, Unnormalize
 from lerobot.policies.pretrained import PreTrainedPolicy
 from lerobot.utils.random_utils import seeded_context
-from lerobot.configs.default import DatasetConfig
-from lerobot.configs.train import TrainPipelineConfig
-from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
 from tests.artifacts.policies.save_policy_to_safetensors import get_policy_stats
 from tests.utils import DEVICE, require_cpu, require_env, require_x86_64_kernel