From 1e049fbef7bac4e07793b95c1a279f10f26f7f37 Mon Sep 17 00:00:00 2001 From: Bryson Jones Date: Mon, 15 Dec 2025 21:04:13 -0800 Subject: [PATCH] add test handling for multitask dit when transformers isnt available --- tests/policies/multi_task_dit/test_multi_task_dit.py | 2 ++ 1 file changed, 2 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 a371d0bc8..b4b4c104d 100644 --- a/tests/policies/multi_task_dit/test_multi_task_dit.py +++ b/tests/policies/multi_task_dit/test_multi_task_dit.py @@ -24,6 +24,8 @@ import pytest import torch from torch import Tensor +pytest.importorskip("transformers") + 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