skip if no transformers

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

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

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

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