From 25ecd16b674a6a86fc5e223bdd6ce0daf36dc18a Mon Sep 17 00:00:00 2001 From: Bryson Jones Date: Mon, 15 Dec 2025 21:52:42 -0800 Subject: [PATCH] skip tests without transformers --- tests/policies/multi_task_dit/test_multi_task_dit.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/policies/multi_task_dit/test_multi_task_dit.py b/tests/policies/multi_task_dit/test_multi_task_dit.py index b4b4c104d..5b70422d4 100644 --- a/tests/policies/multi_task_dit/test_multi_task_dit.py +++ b/tests/policies/multi_task_dit/test_multi_task_dit.py @@ -14,18 +14,27 @@ # See the License for the specific language governing permissions and # limitations under the License. +# ruff: noqa: E402 + """Test script for Multi-Task DiT policy. To run tests locally: python -m pytest tests/policies/multi_task_dit/test_multi_task_dit.py -v """ +import os + import pytest import torch from torch import Tensor pytest.importorskip("transformers") +pytestmark = pytest.mark.skipif( + os.environ.get("CI") == "true" or os.environ.get("GITHUB_ACTIONS") == "true", + reason="This test requires local transformers installation and is not meant for CI", +) + from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature from lerobot.policies.multi_task_dit.configuration_multi_task_dit import MultiTaskDiTConfig from lerobot.policies.multi_task_dit.modeling_multi_task_dit import MultiTaskDiTPolicy