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_pi0_openpi.py
+4
View File
@@ -2,8 +2,12 @@
"""Test script to verify PI0OpenPI policy integration with LeRobot."""
import pytest
import torch
# Skip entire module if transformers is not available
pytest.importorskip("transformers")
from lerobot.policies.factory import make_policy_config
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy
from tests.utils import require_nightly_gpu