From d074231e088c691a0124e5130ea58bb9f284c38a Mon Sep 17 00:00:00 2001
From: Isaac Chung <chungisaac1217@gmail.com>
Date: Tue, 13 Aug 2024 09:03:39 +0000
Subject: [PATCH] directly pass in bm25

---
 models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models.py b/models.py
index 21dfc51..6c3ca66 100644
--- a/models.py
+++ b/models.py
@@ -223,7 +223,7 @@ def retrieve(self, query, corpus, model_name, topk=1):
         corpus_format = CORPUS_TO_FORMAT[corpus]
 
         if "BM25" in model_name:
-            index = self.load_bm25_index(model_name, corpus)
+            index = self.load_bm25_index("BM25", corpus)
             docs = index.search([query], topk=topk)
             if corpus == "stackexchange":
                 return [[query, corpus_format.format(text=docs[0][0]["text"])]]