fix: Optimum - do not explictly pin transformers (#1303)

* pin transformers!=4.48.0 * lint * try removing explicit transformers dep
deepset-ai · Jan 20, 2025 · ae27793 · ae27793
1 parent e973a19
commit ae27793
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/integrations/optimum/pyproject.toml b/integrations/optimum/pyproject.toml
@@ -27,7 +27,6 @@ classifiers = [
 ]
 dependencies = [
   "haystack-ai",
-  "transformers[sentencepiece]",
   # The main export function of Optimum into ONNX has hidden dependencies.
   # It depends on either "sentence-transformers", "diffusers" or "timm", based
   # on which model is loaded from HF Hub.

diff --git a/integrations/optimum/tests/test_optimum_document_embedder.py b/integrations/optimum/tests/test_optimum_document_embedder.py
@@ -99,7 +99,9 @@ def test_init_with_parameters(self, mock_check_valid_model):  # noqa: ARG002
         assert embedder._backend.parameters.optimizer_settings is None
         assert embedder._backend.parameters.quantizer_settings is None
 
-    def test_to_and_from_dict(self, mock_check_valid_model, mock_get_pooling_mode):  # noqa: ARG002
+    def test_to_and_from_dict(self, mock_check_valid_model, mock_get_pooling_mode, monkeypatch):  # noqa: ARG002
+        monkeypatch.delenv("HF_API_TOKEN", raising=False)
+        monkeypatch.delenv("HF_TOKEN", raising=False)
         component = OptimumDocumentEmbedder()
         data = component.to_dict()
 

diff --git a/integrations/optimum/tests/test_optimum_text_embedder.py b/integrations/optimum/tests/test_optimum_text_embedder.py
@@ -84,7 +84,9 @@ def test_init_with_parameters(self, mock_check_valid_model):  # noqa: ARG002
         assert embedder._backend.parameters.optimizer_settings is None
         assert embedder._backend.parameters.quantizer_settings is None
 
-    def test_to_and_from_dict(self, mock_check_valid_model, mock_get_pooling_mode):  # noqa: ARG002
+    def test_to_and_from_dict(self, mock_check_valid_model, mock_get_pooling_mode, monkeypatch):  # noqa: ARG002
+        monkeypatch.delenv("HF_API_TOKEN", raising=False)
+        monkeypatch.delenv("HF_TOKEN", raising=False)
         component = OptimumTextEmbedder()
         data = component.to_dict()