From cfa13c4fa9e36d829d4d9531f28f6a01191a6b24 Mon Sep 17 00:00:00 2001 From: "Luis Felipe A. Venezian" <36893796+luisvenezian@users.noreply.github.com> Date: Sun, 6 Apr 2025 10:02:02 -0300 Subject: [PATCH] only update the results dict when query scores are fully calculated --- beir/retrieval/search/lexical/bm25_search.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/beir/retrieval/search/lexical/bm25_search.py b/beir/retrieval/search/lexical/bm25_search.py index 2eff4370..a7d70be1 100644 --- a/beir/retrieval/search/lexical/bm25_search.py +++ b/beir/retrieval/search/lexical/bm25_search.py @@ -82,7 +82,8 @@ def search( for corpus_id, score in hit["hits"]: if corpus_id != query_id: # query doesnt return in results scores[corpus_id] = score - self.results[query_id] = scores + + self.results[query_id] = scores return self.results