silent linter in xvlatest

This commit is contained in:
Jade Choghari
2025-11-25 15:08:19 +01:00
parent 066fb1bd5d
commit 829428ac81
@@ -15,6 +15,7 @@
# limitations under the License. # limitations under the License.
"""Test script to verify XVLA policy integration with LeRobot vs the original implementation, only meant to be run locally!""" """Test script to verify XVLA policy integration with LeRobot vs the original implementation, only meant to be run locally!"""
# ruff: noqa: E402
import gc import gc
import os import os
@@ -26,10 +27,6 @@ import numpy as np
import pytest import pytest
import torch import torch
from lerobot.policies.xvla.configuration_xvla import XVLAConfig
from lerobot.policies.xvla.modeling_xvla import XVLAPolicy
from lerobot.policies.xvla.processor_xvla import make_xvla_pre_post_processors
# Skip if transformers is not available # Skip if transformers is not available
pytest.importorskip("transformers") pytest.importorskip("transformers")
@@ -39,6 +36,9 @@ pytestmark = pytest.mark.skipif(
reason="This test requires XVLA model access and is not meant for CI", reason="This test requires XVLA model access and is not meant for CI",
) )
from lerobot.policies.xvla.configuration_xvla import XVLAConfig
from lerobot.policies.xvla.modeling_xvla import XVLAPolicy
from lerobot.policies.xvla.processor_xvla import make_xvla_pre_post_processors
from lerobot.processor import PolicyAction, PolicyProcessorPipeline # noqa: E402 from lerobot.processor import PolicyAction, PolicyProcessorPipeline # noqa: E402
from lerobot.utils.constants import OBS_IMAGES, OBS_STATE # noqa: E402 from lerobot.utils.constants import OBS_IMAGES, OBS_STATE # noqa: E402