Commit 8b038c2a authored by Leo Gao's avatar Leo Gao
Browse files

Update spacing

parent 3a90e246
......@@ -39,13 +39,13 @@ class Winogrande(HFTask):
return doc["sentence"][:pronoun_loc] + option
def doc_to_target(self, doc):
return " " + self.partial_target(doc)
return self.partial_target(doc)
@classmethod
def partial_target(cls, doc):
# The target is everything after the document specified pronoun.
pronoun_loc = doc["sentence"].index("_") + 1
return doc["sentence"][pronoun_loc:].strip()
return " " + doc["sentence"][pronoun_loc:].strip()
def construct_requests(self, doc, ctx):
"""Uses RequestFactory to construct Requests and returns an iterable of
......
......@@ -70,13 +70,13 @@ class WinogradSchemaChallenge273(HFTask):
return doc["text"][:doc["pronoun_loc"]] + option
def doc_to_target(self, doc):
return " " + self.partial_target(doc)
return self.partial_target(doc)
@classmethod
def partial_target(cls, doc):
# The target is everything after the document specified pronoun.
start_index = doc["pronoun_loc"] + len(doc["pronoun"])
return doc["text"][start_index:].strip()
return " " + doc["text"][start_index:].strip()
def construct_requests(self, doc, ctx):
"""Uses RequestFactory to construct Requests and returns an iterable of
......
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