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

replace decorator test_pi0_openpi

Browse Source
This commit is contained in:
Pepijn
2025-09-17 16:04:20 +02:00
parent c3b2fadb35
commit 8c0cdb00a6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_pi0_openpi.py
+3 -3
View File
@@ -10,10 +10,10 @@ pytest.importorskip("transformers")
from lerobot.policies.factory import make_policy_config from lerobot.policies.factory import make_policy_config
from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy from lerobot.policies.pi0_openpi import PI0OpenPIConfig, PI0OpenPIPolicy
from tests.utils import require_nightly_gpu from tests.utils import require_cuda
@require_nightly_gpu @require_cuda
def test_policy_instantiation(): def test_policy_instantiation():
"""Test basic policy instantiation.""" """Test basic policy instantiation."""
print("Testing PI0OpenPI policy instantiation...") print("Testing PI0OpenPI policy instantiation...")
@@ -67,7 +67,7 @@ def test_policy_instantiation():
raise raise
@require_nightly_gpu @require_cuda
def test_config_creation(): def test_config_creation():
"""Test policy config creation through factory.""" """Test policy config creation through factory."""
print("\nTesting config creation through factory...") print("\nTesting config creation through factory...")