Update reranker/rankllm.py

reranker/rankllm.py

@@ -283,18 +283,24 @@ class RankLLM(ABC):
             the permuted items.
         """
         response = self._clean_response(permutation)
+        print(f"response after cleaning: {response}")
         response = [int(x) - 1 for x in response.split()]
+        print(f"response after splitting: {response}")
         response = self._remove_duplicate(response)
+        print(f"response after deduplication: {response}")
         cut_range = copy.deepcopy(result.hits[rank_start:rank_end])
         original_rank = [tt for tt in range(len(cut_range))]
         response = [ss for ss in response if ss in original_rank]
+        print(f"response after selection: {response}")
         response = response + [tt for tt in original_rank if tt not in response]
+        print(f"response after appending all original: {response}")
         for j, x in enumerate(response):
             result.hits[j + rank_start] = copy.deepcopy(cut_range[x])
-            if "rank" in result.hits[j + rank_start]:
-                result.hits[j + rank_start]["rank"] = cut_range[j]["rank"]
-            if "score" in result.hits[j + rank_start]:
-                result.hits[j + rank_start]["score"] = cut_range[j]["score"]
+            
+            # if "rank" in result.hits[j + rank_start]:
+            #     result.hits[j + rank_start]["rank"] = cut_range[j]["rank"]
+            # if "score" in result.hits[j + rank_start]:
+            #     result.hits[j + rank_start]["score"] = cut_range[j]["score"]
         return result
 
     def parse_reasoning_permutation(self, response: str) -> Tuple[str, bool]: