mrinjera commited on
Commit
0bc5e4e
·
verified ·
1 Parent(s): cc568d3

chore: testing receive_permutation in reranker/rankllm.py

Browse files
Files changed (1) hide show
  1. 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(