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

skip if no transformers

Browse Source
This commit is contained in:
Pepijn
2025-09-16 18:34:52 +02:00
parent 8f624f1c1e
commit 3a6a2dc637
4 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_pi0_pi05_hub.py
+3
View File
@@ -5,6 +5,9 @@
import pytest
import torch
# Skip entire module if transformers is not available
pytest.importorskip("transformers")
from lerobot.policies.pi0_openpi import PI0OpenPIPolicy
from lerobot.policies.pi05_openpi.modeling_pi05openpi import PI05OpenPIPolicy
from tests.utils import require_nightly_gpu