fix tests

tests/datasets/test_datasets.py

@@ -25,9 +25,6 @@ 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_pi0_original_vs_lerobot.py

@@ -5,8 +5,9 @@ import os
 import pytest
 import torch
 
-# Skip entire module if openpi is not available
+# Skip entire module if openpi or transformers is not available
 pytest.importorskip("openpi")
+pytest.importorskip("transformers")
 
 from openpi.models_pytorch import preprocessing_pytorch as openpi_preprocessing