From 8c0cdb00a60615d5edc1c97be15c4b8455649148 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Wed, 17 Sep 2025 16:04:20 +0200 Subject: [PATCH] replace decorator test_pi0_openpi --- tests/policies/test_pi0_openpi.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/policies/test_pi0_openpi.py b/tests/policies/test_pi0_openpi.py index b89a78175..966323aba 100644 --- a/tests/policies/test_pi0_openpi.py +++ b/tests/policies/test_pi0_openpi.py @@ -10,10 +10,10 @@ 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 +from tests.utils import require_cuda -@require_nightly_gpu +@require_cuda def test_policy_instantiation(): """Test basic policy instantiation.""" print("Testing PI0OpenPI policy instantiation...") @@ -67,7 +67,7 @@ def test_policy_instantiation(): raise -@require_nightly_gpu +@require_cuda def test_config_creation(): """Test policy config creation through factory.""" print("\nTesting config creation through factory...")