From 69b1f7b11810f82529394c32e822c1794b13d0a1 Mon Sep 17 00:00:00 2001
From: Michel Aractingi <michel.aractingi@huggingface.co>
Date: Wed, 2 Jul 2025 18:20:01 +0200
Subject: [PATCH] nit precommit

---
 tests/fixtures/dataset_factories.py | 2 +-
 tests/policies/test_policies.py     | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py
index 5e1a4e2a4..3f93b3d65 100644
--- a/tests/fixtures/dataset_factories.py
+++ b/tests/fixtures/dataset_factories.py
@@ -26,7 +26,6 @@ import pytest
 import torch
 from datasets import Dataset
 
-from lerobot.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,
@@ -39,6 +38,7 @@ from lerobot.datasets.utils import (
     get_hf_features_from_features,
     hf_transform_to_torch,
 )
+from lerobot.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