Reformat

castorini · Sep 27, 2024 · 329d6eb · 329d6eb
1 parent 9e7f846
commit 329d6eb
Showing 1 changed file with 5 additions and 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
@@ -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)