fix tests

This commit is contained in:
Pepijn
2025-09-16 19:03:01 +02:00
parent 3a6a2dc637
commit ac5f77ccbb
5 changed files with 4 additions and 44 deletions
@@ -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