From 316f16630f467a9dd077dff601074994cd67dccd Mon Sep 17 00:00:00 2001 From: Connor Shorten Date: Thu, 27 Nov 2025 16:41:39 +0000 Subject: [PATCH] Update ce_ranker.py --- ce_ranker.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/ce_ranker.py b/ce_ranker.py index 49a9a04..e848909 100644 --- a/ce_ranker.py +++ b/ce_ranker.py @@ -59,10 +59,8 @@ class CERankerAgent(PrecompiledAgent): *[self._score_document(query, doc) for doc in documents] ) - # Sort by the score (descending). If score is True, it comes first. scored_results.sort(key=lambda x: x[1], reverse=True) - # Return a list of documents (not joined as a string), up to k results return [doc for doc, score in scored_results[:k]] def __call__(self, query: str, k: int | None = None) -> list[str]: