fix tets

huggingface · Oct 11, 2024 · 6f7c6c5 · 6f7c6c5
1 parent 3739f07
commit 6f7c6c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/exporters/onnx/test_exporters_onnx_cli.py b/tests/exporters/onnx/test_exporters_onnx_cli.py
@@ -602,6 +602,7 @@ def test_diffusion(self):
                 check=True,
             )
 
+    @require_sentence_transformers
     def test_sentence_transformers(self):
         with TemporaryDirectory() as tmpdirname:
             subprocess.run(