Upload nlp_pipeline.py

toxra_core/nlp_pipeline.py

@@ -256,7 +256,7 @@ def hybrid_rank_text_items(
 
     rrf = _rrf_score(rank_lists)
     final_order = sorted(rrf.keys(), key=lambda idx: rrf[idx], reverse=True)
-    selected_indices = final_order[: max(1, int(top_k))]
+    selected_indices = [int(x) for x in final_order[: max(1, int(top_k))]]
 
     selected: List[Dict[str, Any]] = []
     for idx in selected_indices:
@@ -273,7 +273,7 @@ def hybrid_rank_text_items(
 
     diagnostics = {
         "ranking_method": method,
-        "selected_indices": selected_indices,
+        "selected_indices": [int(x) for x in selected_indices],
         "coverage_by_query_family": cov,
         "coverage_score": cov_score,
         "component_scores": {