From 329d6eb4dae6b1eef3e8a2d08afa6dd5598bda7c Mon Sep 17 00:00:00 2001 From: Yidi Chen Date: Fri, 27 Sep 2024 01:04:28 +0000 Subject: [PATCH] Reformat --- .../rerank/listwise/reorder/top_down_reorder_policy.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/rank_llm/rerank/listwise/reorder/top_down_reorder_policy.py b/src/rank_llm/rerank/listwise/reorder/top_down_reorder_policy.py index f13df2c..4b8709f 100644 --- a/src/rank_llm/rerank/listwise/reorder/top_down_reorder_policy.py +++ b/src/rank_llm/rerank/listwise/reorder/top_down_reorder_policy.py @@ -107,8 +107,11 @@ def perform(self): yield requests - for request, request_indices, i \ - in zip(requests, req_inds, range(window_size, len(indices), window_size - 1)): + for request, request_indices, i in zip( + requests, + req_inds, + range(window_size, len(indices), window_size - 1), + ): request_indices = [ request_indices[i] for i in self._unpad(request_indices, request.result)