Commit 2078adba authored by Leo Gao's avatar Leo Gao
Browse files

Make CoQA take the first line of output excluding any leading whitespace

parent cc474a94
......@@ -115,7 +115,7 @@ class CoQA(Task):
language description, as well as the few shot examples, and the question
part of the document for `doc`.
"""
cont_request = rf.greedy_until(ctx, ['\nQ:'])
cont_request = rf.greedy_until(ctx, ['\nQ:']).strip().split('\n')[0]
return cont_request
def process_results(self, doc, results):
......
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