Small typos

d51f3e7f · Charles Foster · 9291edaa · d51f3e7f
Commit d51f3e7f authored Oct 24, 2020 by Charles Foster
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

lm_eval/tasks/naturalqs.py lm_eval/tasks/naturalqs.py +1 -2

No files found.
--- a/lm_eval/tasks/naturalqs.py
+++ b/lm_eval/tasks/naturalqs.py
@@ -20,9 +20,8 @@ class NaturalQs(HFTask):
    def training_docs(self):
        # Cache training for faster few-shot.
        # Data is too large to fit in memory.
-        self.data["train"])

-        return self.data["train"])
+        return self.data["train"]

    def doc_to_text(self, doc, include_target=True):
        question = doc['question']['text']