Fix key name of retrieved_contexts in the dataset evaluation sample (#68

)
relari-ai · Jun 11, 2024 · 31c4c1c · 31c4c1c
1 parent dba8fb8
commit 31c4c1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/evaluation_on_dataset.py b/examples/evaluation_on_dataset.py
@@ -14,11 +14,11 @@
     dataset=dataset,
     eval=[
         PrecisionRecallF1().use(
-            retrieved_context=dataset.retrieved_context,  # type: ignore
+            retrieved_context=dataset.retrieved_contexts,  # type: ignore
             ground_truth_context=dataset.ground_truth_contexts,  # type: ignore
         ),
         RankedRetrievalMetrics().use(
-            retrieved_context=dataset.retrieved_context,  # type: ignore
+            retrieved_context=dataset.retrieved_contexts,  # type: ignore
             ground_truth_context=dataset.ground_truth_contexts,  # type: ignore
         ),
     ],