Skip to content

GitLab

Commit 1c9432de authored by Leo Gao's avatar Leo Gao
Browse files

Fix MNLI train set

parent 5076d212
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
lm_eval/tasks/glue.py
View file @ 1c9432de
...@@ -76,6 +76,10 @@ class MNLI(HFTask): ...@@ -76,6 +76,10 @@ class MNLI(HFTask):
def has_test_docs(self): def has_test_docs(self):
return True return True
def training_docs(self):
if self.has_validation_docs():
return self.data["train_matched"]
def validation_docs(self): def validation_docs(self):
if self.has_validation_docs(): if self.has_validation_docs():
return self.data["validation_matched"] return self.data["validation_matched"]
......
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