Skip to content

GitLab

Unverified Commit 8c831939 authored by sdtblck's avatar sdtblck Committed by GitHub
Browse files

Update mlqa.py

parent 6cea1de2
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
lm_eval/tasks/mlqa.py
View file @ 8c831939
...@@ -232,6 +232,9 @@ class MLQABase(HFTask): ...@@ -232,6 +232,9 @@ class MLQABase(HFTask):
""" """
continuation = results continuation = results
if isinstance(continuation, list):
continuation = continuation[0]
predictions = { predictions = {
'id': doc['id'], 'id': doc['id'],
'prediction_text': continuation, 'prediction_text': continuation,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment