chore: testing receive_permutation in reranker/rankllm.py
Browse files- reranker/rankllm.py +5 -0
reranker/rankllm.py
CHANGED
|
@@ -133,6 +133,11 @@ class RankLLM(ABC):
|
|
| 133 |
result.ranking_exec_summary.append(ranking_exec_info)
|
| 134 |
|
| 135 |
result = self.receive_permutation(result, permutation, rank_start, rank_end)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 136 |
return result
|
| 137 |
|
| 138 |
def permutation_pipeline_batched(
|
|
|
|
| 133 |
result.ranking_exec_summary.append(ranking_exec_info)
|
| 134 |
|
| 135 |
result = self.receive_permutation(result, permutation, rank_start, rank_end)
|
| 136 |
+
|
| 137 |
+
prompt, in_token_count = self.create_prompt(result, rank_start, rank_end)
|
| 138 |
+
if logging:
|
| 139 |
+
print(f"After receiving permutation: {prompt}")
|
| 140 |
+
|
| 141 |
return result
|
| 142 |
|
| 143 |
def permutation_pipeline_batched(
|