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

add comment pi0 pi05 tests, add image features to pi0 pi05 hub tests

Browse Source
This commit is contained in:
Pepijn
2025-09-17 20:04:51 +02:00
parent 6467ce10d4
commit 02f52807e6
3 changed files with 50 additions and 213 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_pi0_openpi.py
+11 -6
View File
@@ -1,16 +1,21 @@
#!/usr/bin/env python
"""Test script to verify PI0OpenPI policy integration with LeRobot."""
"""Test script to verify PI0OpenPI policy integration with LeRobot, only meant to be run locally!"""
import os
import pytest
import torch
# Skip entire module if transformers is not available
pytest.importorskip("transformers")
# Skip this entire module in CI
pytestmark = pytest.mark.skipif(
os.environ.get("CI") == "true" or os.environ.get("GITHUB_ACTIONS") == "true",
reason="This test requires local OpenPI installation and is not meant for CI",
)
from lerobot.policies.factory import make_policy_config
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy
from tests.utils import require_cuda
from lerobot.policies.factory import make_policy_config # noqa: E402
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy # noqa: E402
from tests.utils import require_cuda # noqa: E402
@require_cuda