Fix issue introduced in merge

UKPLab · Oct 17, 2024 · 2cfd817 · 2cfd817
1 parent f8b4b4e
commit 2cfd817
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sentence_transformers/evaluation/InformationRetrievalEvaluator.py b/sentence_transformers/evaluation/InformationRetrievalEvaluator.py
@@ -331,8 +331,8 @@ def compute_metrices(
                 ):
                     sub_corpus_embeddings = corpus_model.encode(
                         self.corpus[corpus_start_idx:corpus_end_idx],
-                        prompt_name=self.query_prompt_name,
-                        prompt=self.query_prompt,
+                        prompt_name=self.corpus_prompt_name,
+                        prompt=self.corpus_prompt,
                         batch_size=self.batch_size,
                         show_progress_bar=False,
                         convert_to_tensor=True,