Remove import of the non-existent `NLP_TASK` module

ddc044eb · Jonathan Tow · 9a9c8e3c · ddc044eb
Commit ddc044eb authored Mar 27, 2021 by Jonathan Tow
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

lm_eval/tasks/wikitext.py lm_eval/tasks/wikitext.py +3 -3

No files found.
--- a/lm_eval/tasks/wikitext.py
+++ b/lm_eval/tasks/wikitext.py
-from . common import NLP_TASK
+from . common import HFTask


-class WikiText103(NLP_TASK):
+class WikiText103(HFTask):
    NLP_PATH = "wikitext"
    NLP_NAME = "wikitext-103-raw-v1"

@@ -63,7 +63,7 @@ class WikiText103(NLP_TASK):
        raise NotImplementedError('Evaluation not implemented')


-class WikiText2(NLP_TASK):
+class WikiText2(HFTask):
    NLP_PATH = "wikitext"
    NLP_NAME = "wikitext-2-raw-v1"