From d5d9971bdc14daba8ec4e1bef4c93e65b70c7bf7 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Wed, 12 Jun 2024 13:44:50 +0800
Subject: [PATCH] fix: add skip
---
tests/unit/encoders/test_clip.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/tests/unit/encoders/test_clip.py b/tests/unit/encoders/test_clip.py
index 5225cf4a..1cfc7d43 100644
--- a/tests/unit/encoders/test_clip.py
+++ b/tests/unit/encoders/test_clip.py
@@ -34,14 +34,20 @@ def misshaped_pil_image():
class TestClipEncoder:
- def test_clip_encoder__import_errors_transformers(self, mocker):
+ @pytest.mark.skipif(
+ os.environ.get("RUN_HF_TESTS") is None, reason="Set RUN_HF_TESTS=1 to run"
+ )
+ def test_clip_encoder__import_errors_transformers(self):
with patch.dict("sys.modules", {"transformers": None}):
with pytest.raises(ImportError) as error:
CLIPEncoder()
assert "install transformers" in str(error.value)
- def test_clip_encoder__import_errors_torch(self, mocker):
+ @pytest.mark.skipif(
+ os.environ.get("RUN_HF_TESTS") is None, reason="Set RUN_HF_TESTS=1 to run"
+ )
+ def test_clip_encoder__import_errors_torch(self):
with patch.dict("sys.modules", {"torch": None}):
with pytest.raises(ImportError) as error:
CLIPEncoder()
--
GitLab