Skip to content

Commit 01957b1

Browse files
authored
refactor: use token inSentenceTransformersDiversityRanker (#9126)
* chore: use token in SASEvaluator and SentenceTransformersDiversityRanker * directly pass token to CrossEncoder * restrict scope to SentenceTransformersDiversityRanker
1 parent 657d09d commit 01957b1

File tree

3 files changed

+10
-5
lines changed

3 files changed

+10
-5
lines changed

haystack/components/rankers/sentence_transformers_diversity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ def warm_up(self):
199199
self.model = SentenceTransformer(
200200
model_name_or_path=self.model_name_or_path,
201201
device=self.device.to_torch_str(),
202-
use_auth_token=self.token.resolve_value() if self.token else None,
202+
token=self.token.resolve_value() if self.token else None,
203203
model_kwargs=self.model_kwargs,
204204
tokenizer_kwargs=self.tokenizer_kwargs,
205205
config_kwargs=self.config_kwargs,
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
enhancements:
3+
- |
4+
Updated `SentenceTransformersDiversityRanker` to use the `token` parameter internally instead
5+
of the deprecated `use_auth_token`. The public API of this component already utilizes `token`.

test/components/rankers/test_sentence_transformers_diversity.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,7 @@ def test_warm_up(self, similarity, monkeypatch):
290290
mock_model_class.assert_called_once_with(
291291
model_name_or_path="mock_model_name",
292292
device=ComponentDevice.resolve_device(None).to_torch_str(),
293-
use_auth_token=None,
293+
token=None,
294294
model_kwargs=None,
295295
tokenizer_kwargs=None,
296296
config_kwargs=None,
@@ -740,7 +740,7 @@ def test_model_onnx_backend(self, mocked_sentence_transformer):
740740
mocked_sentence_transformer.assert_called_once_with(
741741
model_name_or_path="sentence-transformers/all-MiniLM-L6-v2",
742742
device="cpu",
743-
use_auth_token=None,
743+
token=None,
744744
model_kwargs={"file_name": "onnx/model.onnx"},
745745
tokenizer_kwargs=None,
746746
config_kwargs=None,
@@ -761,7 +761,7 @@ def test_model_openvino_backend(self, mocked_sentence_transformer):
761761
mocked_sentence_transformer.assert_called_once_with(
762762
model_name_or_path="sentence-transformers/all-MiniLM-L6-v2",
763763
device="cpu",
764-
use_auth_token=None,
764+
token=None,
765765
model_kwargs={"file_name": "openvino/openvino_model.xml"},
766766
tokenizer_kwargs=None,
767767
config_kwargs=None,
@@ -782,7 +782,7 @@ def test_dtype_on_gpu(self, mocked_sentence_transformer, model_kwargs):
782782
mocked_sentence_transformer.assert_called_once_with(
783783
model_name_or_path="sentence-transformers/all-MiniLM-L6-v2",
784784
device="cuda:0",
785-
use_auth_token=None,
785+
token=None,
786786
model_kwargs=model_kwargs,
787787
tokenizer_kwargs=None,
788788
config_kwargs=None,

0 commit comments

Comments
 (0)