Update retriever.py

retriever.py

@@ -112,9 +112,9 @@ class Retriever:
         # Sort the documents by their BM25 scores in descending order
         sorted_doc_indices = np.argsort(scores)
 
-        print("Score:", scores[sorted_doc_indices[-i]] )
-        print(self.docs[sorted_doc_indices[-i]])
-        print("Doc number:", sorted_doc_indices[-i])
+        print("Score:", scores[sorted_doc_indices[-1]] )
+        print(self.docs[sorted_doc_indices[-1]])
+        print("Doc number:", sorted_doc_indices[-1])
 
         result_docs = [self.docs[i] for i in sorted_doc_indices[-n:] if scores[i] > 0]