From b5596289a4a20b499eadf6b1592548006fb49c83 Mon Sep 17 00:00:00 2001 From: Ian Thompson Date: Tue, 1 Oct 2024 10:21:30 -0500 Subject: [PATCH] modified: sentence_transformers/cross_encoder/CrossEncoder.py - Changed method of unioning dictionary to `update` in :method:`CrossEncoder.rank` to avoid unpacking original dictionary. --- sentence_transformers/cross_encoder/CrossEncoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sentence_transformers/cross_encoder/CrossEncoder.py b/sentence_transformers/cross_encoder/CrossEncoder.py index e30024f87..c8d8d1d84 100644 --- a/sentence_transformers/cross_encoder/CrossEncoder.py +++ b/sentence_transformers/cross_encoder/CrossEncoder.py @@ -546,7 +546,7 @@ def rank( for i, score in enumerate(scores): results.append({"corpus_id": i, "score": score}) if return_documents: - results[-1] = {**results[-1], "text": documents[i]} + results[-1].update({"text": documents[i]}) results = sorted(results, key=lambda x: x["score"], reverse=True) return results[:top_k]