Commit d1319950 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Add newline to end of file

parent 09cd76c7
...@@ -9,7 +9,7 @@ def test_description_dict(): ...@@ -9,7 +9,7 @@ def test_description_dict():
task_names = ["hellaswag", "winogrande"] task_names = ["hellaswag", "winogrande"]
description_dict = { description_dict = {
"hellaswag": "Label for the relevant action:\nSentences describing context, with an incomplete sentence trailing answer that plausibly completes the situation.", "hellaswag": "Label for the relevant action:\nSentences describing context, with an incomplete sentence trailing answer that plausibly completes the situation.",
"winogrande": "Winograd schema sentence including a either a ___ blank with a missing word, making the pronoun ambiguous, or the same with the word filled in." "winogrande": "Winograd schema sentence including a either a ___ blank with a missing word, making the pronoun ambiguous, or the same with the word filled in.",
} }
task_dict = lm_eval.tasks.get_task_dict(task_names) task_dict = lm_eval.tasks.get_task_dict(task_names)
...@@ -31,9 +31,7 @@ def test_description_dict(): ...@@ -31,9 +31,7 @@ def test_description_dict():
) )
for _, doc in ( for _, doc in (
zip(range(num_examples), docs) zip(range(num_examples), docs) if num_examples > 0 else enumerate(docs)
if num_examples > 0
else enumerate(docs)
): ):
ctx = task.fewshot_context( ctx = task.fewshot_context(
doc=doc, doc=doc,
...@@ -43,4 +41,6 @@ def test_description_dict(): ...@@ -43,4 +41,6 @@ def test_description_dict():
description=description, description=description,
) )
assert description in ctx assert description in ctx
test_description_dict()
\ No newline at end of file
test_description_dict()
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