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