Commit da016758 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Revert "Update `doc_to_text` to show candidate fill-ins"

This reverts commit 5d431ba9.
parent 5d431ba9
import re
import numpy as np
from ..base import rf, mean
from . common import HFTask
......@@ -35,9 +34,7 @@ class HellaSwag(HFTask):
"plausibly completes the situation."
def doc_to_text(self, doc):
text = doc['activity_label'] + ': ' + doc['ctx'] + '\n'
text += ''.join([ending + "\n" for ending in doc['endings']])
return text + "\n"
return doc['activity_label'] + ': ' + doc['ctx'] + '\n'
def doc_to_target(self, doc):
letter_answer = doc['label']
......
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