fix: patch type issue in evolution parsing (#980)

fixes: #900
explodinggradients · May 21, 2024 · 54e9f4d · 54e9f4d
1 parent 8d658a1
commit 54e9f4d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/ragas/testset/evolutions.py b/src/ragas/testset/evolutions.py
@@ -202,7 +202,13 @@ async def generate_datarow(
             if isinstance(relevant_contexts_result, dict)
             else None
         )
-        if relevant_context_indices is None:
+
+        if relevant_context_indices is not None:
+            relevant_context_indices = [
+                idx for idx in relevant_context_indices if isinstance(idx, int)
+            ]
+
+        if relevant_context_indices is None or not relevant_context_indices:
             relevant_context = CurrentNodes(
                 root_node=current_nodes.root_node, nodes=current_nodes.nodes
             )