Skip to content

Commit 14895f6

Browse files
sjrlanakin87
andauthored
chore: Use token instead of use_auth_token because of deprecation warning (#8552)
* Use token instead of use_auth_token because of deprecation warning * Fix test * pylint * fix linting --------- Co-authored-by: Stefano Fiorucci <[email protected]>
1 parent bd77120 commit 14895f6

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

haystack/components/embedders/backends/sentence_transformers_backend.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class _SentenceTransformersEmbeddingBackendFactory:
1919
_instances: Dict[str, "_SentenceTransformersEmbeddingBackend"] = {}
2020

2121
@staticmethod
22-
def get_embedding_backend(
22+
def get_embedding_backend( # pylint: disable=too-many-positional-arguments
2323
model: str,
2424
device: Optional[str] = None,
2525
auth_token: Optional[Secret] = None,
@@ -52,7 +52,7 @@ class _SentenceTransformersEmbeddingBackend:
5252
Class to manage Sentence Transformers embeddings.
5353
"""
5454

55-
def __init__(
55+
def __init__( # pylint: disable=too-many-positional-arguments
5656
self,
5757
model: str,
5858
device: Optional[str] = None,
@@ -67,7 +67,7 @@ def __init__(
6767
self.model = SentenceTransformer(
6868
model_name_or_path=model,
6969
device=device,
70-
use_auth_token=auth_token.resolve_value() if auth_token else None,
70+
token=auth_token.resolve_value() if auth_token else None,
7171
trust_remote_code=trust_remote_code,
7272
truncate_dim=truncate_dim,
7373
model_kwargs=model_kwargs,

test/components/embedders/test_sentence_transformers_embedding_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def test_model_initialization(mock_sentence_transformer):
3737
mock_sentence_transformer.assert_called_once_with(
3838
model_name_or_path="model",
3939
device="cpu",
40-
use_auth_token="fake-api-token",
40+
token="fake-api-token",
4141
trust_remote_code=True,
4242
truncate_dim=256,
4343
model_kwargs=None,

0 commit comments

Comments
 (0)