Add missing space in MultiRC

e9e5d0a2 · Leo Gao · 42df113c · e9e5d0a2
Commit e9e5d0a2 authored Feb 09, 2021 by Leo Gao
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/tasks/superglue.py lm_eval/tasks/superglue.py +1 -1

No files found.
--- a/lm_eval/tasks/superglue.py
+++ b/lm_eval/tasks/superglue.py
@@ -218,7 +218,7 @@ class MultiRC(HFTask):
        return f"{doc['paragraph']}\nQuestion: {doc['question']}\nAnswer:"

    def doc_to_target(self, doc):
-        return self.format_answer(answer=doc["answer"], label=doc["label"])
+        return " " + self.format_answer(answer=doc["answer"], label=doc["label"])

    @staticmethod
    def format_answer(answer, label):