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

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_pi05_openpi.py
+4
View File
@@ -2,8 +2,12 @@
"""Test script to verify PI0.5 (pi05) support in PI0OpenPI policy."""
import pytest
import torch
# Skip entire module if transformers is not available
pytest.importorskip("transformers")
from lerobot.policies.pi0_openpi.configuration_pi0openpi import PI0OpenPIConfig
from lerobot.policies.pi0_openpi.modeling_pi0openpi import PI0OpenPIPolicy
from lerobot.policies.pi05_openpi import PI05OpenPIConfig, PI05OpenPIPolicy