admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 12:40:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add test handling for multitask dit when transformers isnt available

Browse Source
This commit is contained in:
Bryson Jones
2025-12-15 21:04:13 -08:00
parent afe2c4d3aa
commit 1e049fbef7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/multi_task_dit/test_multi_task_dit.py
+2
View File
@@ -24,6 +24,8 @@ import pytest
import torch import torch
from torch import Tensor from torch import Tensor
pytest.importorskip("transformers")
from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature 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.configuration_multi_task_dit import MultiTaskDiTConfig
from lerobot.policies.multi_task_dit.modeling_multi_task_dit import MultiTaskDiTPolicy from lerobot.policies.multi_task_dit.modeling_multi_task_dit import MultiTaskDiTPolicy