From 3a6a2dc637a03d76d19657cb5ad36ec83b027489 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 16 Sep 2025 18:34:52 +0200 Subject: [PATCH] skip if no transformers --- tests/datasets/test_datasets.py | 3 +++ tests/policies/test_pi05_openpi.py | 4 ++++ tests/policies/test_pi0_openpi.py | 4 ++++ tests/policies/test_pi0_pi05_hub.py | 3 +++ 4 files changed, 14 insertions(+) diff --git a/tests/datasets/test_datasets.py b/tests/datasets/test_datasets.py index 2eca82346..5e2996a43 100644 --- a/tests/datasets/test_datasets.py +++ b/tests/datasets/test_datasets.py @@ -25,6 +25,9 @@ from huggingface_hub import HfApi from PIL import Image from safetensors.torch import load_file +# Skip entire module if transformers is not available (required by policy factory imports) +pytest.importorskip("transformers") + import lerobot from lerobot.configs.default import DatasetConfig from lerobot.configs.train import TrainPipelineConfig diff --git a/tests/policies/test_pi05_openpi.py b/tests/policies/test_pi05_openpi.py index 477c854b3..8d947e533 100644 --- a/tests/policies/test_pi05_openpi.py +++ b/tests/policies/test_pi05_openpi.py @@ -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 diff --git a/tests/policies/test_pi0_openpi.py b/tests/policies/test_pi0_openpi.py index 36c8200b3..b89a78175 100644 --- a/tests/policies/test_pi0_openpi.py +++ b/tests/policies/test_pi0_openpi.py @@ -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 diff --git a/tests/policies/test_pi0_pi05_hub.py b/tests/policies/test_pi0_pi05_hub.py index 759caf13b..5190b88ed 100644 --- a/tests/policies/test_pi0_pi05_hub.py +++ b/tests/policies/test_pi0_pi05_hub.py @@ -5,6 +5,9 @@ import pytest import torch +# Skip entire module if transformers is not available +pytest.importorskip("transformers") + from lerobot.policies.pi0_openpi import PI0OpenPIPolicy from lerobot.policies.pi05_openpi.modeling_pi05openpi import PI05OpenPIPolicy from tests.utils import require_nightly_gpu