Commit 8888a6be authored by Andy Zou's avatar Andy Zou
Browse files

adjusted naming

parent a6b01474
...@@ -33,7 +33,7 @@ from . import ethics ...@@ -33,7 +33,7 @@ from . import ethics
from . import drop from . import drop
from . import unscramble from . import unscramble
from . import logiqa from . import logiqa
from . import hendrycks from . import hendrycks_test
######################################## ########################################
# Translation tasks # Translation tasks
...@@ -142,7 +142,7 @@ TASK_REGISTRY = { ...@@ -142,7 +142,7 @@ TASK_REGISTRY = {
# e.g. anli, arithmetic, openai_translations, harness_translations # e.g. anli, arithmetic, openai_translations, harness_translations
# hendrycksTest (57 tasks) # hendrycksTest (57 tasks)
**hendrycks.create_all_tasks(), **hendrycks_test.create_all_tasks(),
# e.g. wmt14-fr-en # e.g. wmt14-fr-en
**translation.create_tasks_from_benchmarks(gpt3_translation_benchmarks), **translation.create_tasks_from_benchmarks(gpt3_translation_benchmarks),
......
...@@ -20,10 +20,10 @@ CHOICES = ['A','B','C','D'] ...@@ -20,10 +20,10 @@ CHOICES = ['A','B','C','D']
def create_all_tasks(): def create_all_tasks():
"""Creates a dictionary of tasks from a list of subjects """Creates a dictionary of tasks from a list of subjects
:return: {task_name: task} :return: {task_name: task}
e.g. {hendrycks-abstract_algebra: Task, hendrycks-anatomy: Task} e.g. {hendrycksTest-abstract_algebra: Task, hendrycksTest-anatomy: Task}
""" """
return { return {
f"hendrycks-{sub}": create_task(sub) for sub in SUBJECTS f"hendrycksTest-{sub}": create_task(sub) for sub in SUBJECTS
} }
def create_task(subject): def create_task(subject):
...@@ -40,14 +40,14 @@ class GeneralHendrycksTest(Task): ...@@ -40,14 +40,14 @@ class GeneralHendrycksTest(Task):
def download(self): def download(self):
self.data_dir = "data/hendrycks/" self.data_dir = "data/hendrycksTest/"
if not os.path.exists(self.data_dir): if not os.path.exists(self.data_dir):
sh(""" sh("""
mkdir -p data mkdir -p data
wget https://people.eecs.berkeley.edu/~hendrycks/data.tar -P data/ wget https://people.eecs.berkeley.edu/~hendrycks/data.tar -P data/
tar -xf data/data.tar -C data/ tar -xf data/data.tar -C data/
rm data/data.tar rm data/data.tar
mv data/data data/hendrycks mv data/data data/hendrycksTest
""") """)
def has_training_docs(self): def has_training_docs(self):
......
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