From ad62a2c151e1c4b69104d81fe8fc6aac33202ad8 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Thu, 6 Aug 2026 19:25:12 +0200 Subject: [PATCH] chore: sort imports in vla_jepa tests Pre-existing `I001` violations that `pre-commit run --all-files` auto-fixes. They are unrelated to the documentation work but block the green run the rest of this branch is verified against; confirmed present on main before this branch. Co-Authored-By: Claude Opus 5 --- tests/policies/vla_jepa/test_action_head.py | 1 - tests/policies/vla_jepa/test_configuration.py | 2 +- tests/policies/vla_jepa/test_vla_jepa.py | 1 - 3 files changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/policies/vla_jepa/test_action_head.py b/tests/policies/vla_jepa/test_action_head.py index 5acff63714e..7c48d5646e8 100644 --- a/tests/policies/vla_jepa/test_action_head.py +++ b/tests/policies/vla_jepa/test_action_head.py @@ -16,7 +16,6 @@ make_config, set_seed_all, ) # noqa: E402 - from lerobot.policies.vla_jepa.action_head import ( # noqa: E402 VLAJEPAActionHead, ) diff --git a/tests/policies/vla_jepa/test_configuration.py b/tests/policies/vla_jepa/test_configuration.py index 2eda08ad355..30d6741c3db 100644 --- a/tests/policies/vla_jepa/test_configuration.py +++ b/tests/policies/vla_jepa/test_configuration.py @@ -3,8 +3,8 @@ from __future__ import annotations import pytest -from conftest import ACTION_DIM, ACTION_HORIZON, IMAGE_SIZE, NUM_VIDEO_FRAMES, STATE_DIM, make_config +from conftest import ACTION_DIM, ACTION_HORIZON, IMAGE_SIZE, NUM_VIDEO_FRAMES, STATE_DIM, make_config from lerobot.configs.types import FeatureType, PolicyFeature from lerobot.policies.vla_jepa.configuration_vla_jepa import VLAJEPAConfig from lerobot.utils.constants import ACTION, OBS_IMAGES, OBS_STATE diff --git a/tests/policies/vla_jepa/test_vla_jepa.py b/tests/policies/vla_jepa/test_vla_jepa.py index a3e24a66055..78a732f7f14 100644 --- a/tests/policies/vla_jepa/test_vla_jepa.py +++ b/tests/policies/vla_jepa/test_vla_jepa.py @@ -32,7 +32,6 @@ make_train_batch, set_seed_all, ) - from lerobot.policies.vla_jepa.configuration_vla_jepa import VLAJEPAConfig # noqa: E402 from lerobot.policies.vla_jepa.modeling_vla_jepa import VLAJEPAPolicy # noqa: E402 from lerobot.utils.constants import ACTION # noqa: E402