Commit e6ca2021 authored by ingyuseong's avatar ingyuseong
Browse files

Resolve confilict on init file

parent dae050b4
...@@ -57,6 +57,7 @@ from . import klue ...@@ -57,6 +57,7 @@ from . import klue
from . import ko_translation from . import ko_translation
from . import korquad from . import korquad
from . import korunsmile from . import korunsmile
from . import kohatespeech
######################################## ########################################
# Translation tasks # Translation tasks
...@@ -69,7 +70,6 @@ gpt3_translation_benchmarks = { ...@@ -69,7 +70,6 @@ gpt3_translation_benchmarks = {
"wmt16": ['en-ro', 'ro-en', 'de-en', 'en-de'], # German, Romanian "wmt16": ['en-ro', 'ro-en', 'de-en', 'en-de'], # German, Romanian
} }
# 28 total # 28 total
selected_translation_benchmarks = { selected_translation_benchmarks = {
**gpt3_translation_benchmarks, **gpt3_translation_benchmarks,
...@@ -88,7 +88,7 @@ all_translation_benchmarks = { ...@@ -88,7 +88,7 @@ all_translation_benchmarks = {
# All tasks # All tasks
######################################## ########################################
# task 레지스트리에 클래스 추가
TASK_REGISTRY = { TASK_REGISTRY = {
# GLUE # GLUE
"cola": glue.CoLA, "cola": glue.CoLA,
...@@ -322,9 +322,11 @@ TASK_REGISTRY = { ...@@ -322,9 +322,11 @@ TASK_REGISTRY = {
"ko_en_translation": ko_translation.KoEnTranslation, "ko_en_translation": ko_translation.KoEnTranslation,
"en_ko_translation": ko_translation.EnKoTranslation, "en_ko_translation": ko_translation.EnKoTranslation,
"korunsmile": korunsmile.KorUnSmile "korunsmile": korunsmile.KorUnSmile
"kohatespeech":kohatespeech.HateSpeech,
"kohatespeech_gen_bias":kohatespeech.GenderBias,
"kohatespeech_apeach":kohatespeech.Apeach
} }
ALL_TASKS = sorted(list(TASK_REGISTRY)) ALL_TASKS = sorted(list(TASK_REGISTRY))
......
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