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

adding guards to avoid needing transformers and diffusers for type checking and basic tests

Browse Source
This commit is contained in:
Maximellerbach
2026-05-11 19:05:26 +02:00
parent ff7dc0519e
commit 96a8efddf0
4 changed files with 40 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/vla_jepa/test_vla_jepa.py
+3
View File
@@ -17,6 +17,9 @@ from lerobot.policies.vla_jepa.configuration_vla_jepa import VLAJEPAConfig
from lerobot.policies.vla_jepa.modeling_vla_jepa import VLAJEPAPolicy
from lerobot.utils.constants import ACTION, OBS_IMAGES, OBS_STATE
pytest.importorskip("transformers")
pytest.importorskip("diffusers")
pytestmark = pytest.mark.filterwarnings(
"ignore:In CPU autocast, but the target dtype is not supported:UserWarning"
)