Skip to content

Commit 31c4c1c

Browse files
authored
Fix key name of retrieved_contexts in the dataset evaluation sample (#68)
1 parent dba8fb8 commit 31c4c1c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/evaluation_on_dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,11 @@
1414
dataset=dataset,
1515
eval=[
1616
PrecisionRecallF1().use(
17-
retrieved_context=dataset.retrieved_context, # type: ignore
17+
retrieved_context=dataset.retrieved_contexts, # type: ignore
1818
ground_truth_context=dataset.ground_truth_contexts, # type: ignore
1919
),
2020
RankedRetrievalMetrics().use(
21-
retrieved_context=dataset.retrieved_context, # type: ignore
21+
retrieved_context=dataset.retrieved_contexts, # type: ignore
2222
ground_truth_context=dataset.ground_truth_contexts, # type: ignore
2323
),
2424
],

0 commit comments

Comments
 (0)