Skip to content

GitLab

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

Remove import of the non-existent `NLP_TASK` module

parent 9a9c8e3c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lm_eval/tasks/wikitext.py
View file @ ddc044eb
from . common import NLP_TASK from . common import HFTask
class WikiText103(NLP_TASK): class WikiText103(HFTask):
NLP_PATH = "wikitext" NLP_PATH = "wikitext"
NLP_NAME = "wikitext-103-raw-v1" NLP_NAME = "wikitext-103-raw-v1"
...@@ -63,7 +63,7 @@ class WikiText103(NLP_TASK): ...@@ -63,7 +63,7 @@ class WikiText103(NLP_TASK):
raise NotImplementedError('Evaluation not implemented') raise NotImplementedError('Evaluation not implemented')
class WikiText2(NLP_TASK): class WikiText2(HFTask):
NLP_PATH = "wikitext" NLP_PATH = "wikitext"
NLP_NAME = "wikitext-2-raw-v1" NLP_NAME = "wikitext-2-raw-v1"
......
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