Skip to content

Commit

Permalink
Fix copa ignored docs
Browse files Browse the repository at this point in the history
  • Loading branch information
jon-tow authored and jon-tow committed May 18, 2022
1 parent 4a78c75 commit 6e455eb
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions lm_eval/tasks/superglue.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,17 @@ def training_docs(self):
def validation_docs(self):
return self.dataset["validation"]

def invalid_doc_for_prompt(self, doc) -> bool:
# HACK: Some copa templates have conditionals that ignore documents
# when the condition is not met, like `{if doc['question'] != \"cause\"}`.
# This means the prompt will never produce an input and target.
# TODO: Remove this when fixed in `promptsource`
try:
self.prompt.apply(doc)
return False
except:
return True


# TODO: Check this works with all prompts.
class MultiRC(PromptSourceTask):
Expand Down

0 comments on commit 6e455eb

Please sign in to comment.