Commit 15c186b7 authored by lintangsutawika's avatar lintangsutawika
Browse files

change variable name

parent e5306ea6
...@@ -18,14 +18,14 @@ def t5_prompt_doc_to_text(x): ...@@ -18,14 +18,14 @@ def t5_prompt_doc_to_text(x):
return text return text
def default_doc_to_text(doc): def default_doc_to_text(x):
raw_passage = doc["text"] raw_passage = x["text"]
# NOTE: HuggingFace span indices are word-based not character-based. # NOTE: HuggingFace span indices are word-based not character-based.
pre = " ".join(raw_passage.split()[: doc["span2_index"]]) pre = " ".join(raw_passage.split()[: x["span2_index"]])
post = raw_passage[len(pre) + len(doc["span2_text"]) + 1 :] post = raw_passage[len(pre) + len(x["span2_text"]) + 1 :]
passage = general_detokenize(pre + " *{}*".format(doc["span2_text"]) + post) passage = general_detokenize(pre + " *{}*".format(x["span2_text"]) + post)
noun = doc["span1_text"] noun = x["span1_text"]
pronoun = doc["span2_text"] pronoun = x["span2_text"]
text = ( text = (
f"Passage: {passage}\n" f"Passage: {passage}\n"
+ f'Question: In the passage above, does the pronoun "*{pronoun}*" refer to "*{noun}*"?\n' + f'Question: In the passage above, does the pronoun "*{pronoun}*" refer to "*{noun}*"?\n'
......
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