diff --git a/seacrowd/utils/constants.py b/seacrowd/utils/constants.py index 965701caf..37ef4175a 100644 --- a/seacrowd/utils/constants.py +++ b/seacrowd/utils/constants.py @@ -44,13 +44,9 @@ class Tasks(Enum): WORD_SENSE_DISAMBIGUATION = "WSD" COREFERENCE_RESOLUTION = "COREF" - # Tree CONSTITUENCY_PARSING = "CONST_PAR" - # Single Text Classification - ASPECT_BASED_SENTIMENT_ANALYSIS = "ABSA" - # Single Text Classification (single-label) ABUSIVE_LANGUAGE_PREDICTION = "ABL" DOMAIN_KNOWLEDGE_CLASSIFICATION = "DKC" # classification for non NLP-oriented label @@ -120,6 +116,7 @@ class Tasks(Enum): # ImageText IMAGE_CAPTIONING = "IC" + SIGN_LANGUAGE_RECOGNITION = "SLR" STYLIZED_IMAGE_CAPTIONING = "SIC" VISUALLY_GROUNDED_REASONING = "VGR" @@ -215,21 +212,23 @@ class Licenses(Enum): TASK_TO_SCHEMA = { + Tasks.COREFERENCE_RESOLUTION: "KB", Tasks.DEPENDENCY_PARSING: "KB", Tasks.CONSTITUENCY_PARSING: "TREE", + Tasks.E2E_TASK_ORIENTED_DIALOGUE: "TOD", Tasks.WORD_SENSE_DISAMBIGUATION: "T2T", Tasks.WORD_ANALOGY: "T2T", Tasks.KEYWORD_EXTRACTION: "SEQ_LABEL", - Tasks.COREFERENCE_RESOLUTION: "KB", Tasks.DIALOGUE_SYSTEM: "T2T", - Tasks.E2E_TASK_ORIENTED_DIALOGUE: "TOD", Tasks.KEYWORD_TAGGING: "SEQ_LABEL", Tasks.NAMED_ENTITY_RECOGNITION: "SEQ_LABEL", Tasks.POS_TAGGING: "SEQ_LABEL", Tasks.SENTENCE_ORDERING: "SEQ_LABEL", + Tasks.SPAN_BASED_ABSA: "SEQ_LABEL", Tasks.TOKEN_LEVEL_LANGUAGE_IDENTIFICATION: "SEQ_LABEL", Tasks.COMMONSENSE_REASONING: "QA", Tasks.QUESTION_ANSWERING: "QA", + Tasks.CONCEPT_ALIGNMENT_CLASSIFICATION: "PAIRS", Tasks.NEXT_SENTENCE_PREDICTION: "PAIRS", Tasks.TEXT_RETRIEVAL: "PAIRS", Tasks.TEXTUAL_ENTAILMENT: "PAIRS", @@ -242,13 +241,14 @@ class Licenses(Enum): Tasks.MULTILEXNORM: "T2T", Tasks.SUMMARIZATION: "T2T", Tasks.TRANSLITERATION: "T2T", - Tasks.DOMAIN_KNOWLEDGE_MULTICLASSIFICATION: "TEXT_MULTI", Tasks.ASPECT_BASED_SENTIMENT_ANALYSIS: "TEXT_MULTI", + Tasks.DOMAIN_KNOWLEDGE_MULTICLASSIFICATION: "TEXT_MULTI", Tasks.ABUSIVE_LANGUAGE_PREDICTION: "TEXT", Tasks.DOMAIN_KNOWLEDGE_CLASSIFICATION: "TEXT", Tasks.SENTIMENT_ANALYSIS: "TEXT", Tasks.TAX_COURT_VERDICT: "TEXT", Tasks.EMOTION_CLASSIFICATION: "TEXT", + Tasks.HOAX_NEWS_CLASSIFICATION: "TEXT", Tasks.LANGUAGE_IDENTIFICATION: "TEXT", Tasks.LEGAL_CLASSIFICATION: "TEXT", Tasks.INTENT_CLASSIFICATION: "TEXT", @@ -265,14 +265,11 @@ class Licenses(Enum): Tasks.SPEECH_EMOTION_RECOGNITION: "SPEECH", Tasks.SPEECH_EMOTION_RECOGNITION_MULTILABEL: "SPEECH_MULTI", Tasks.IMAGE_CAPTIONING: "IMTEXT", + Tasks.SIGN_LANGUAGE_RECOGNITION: "IMTEXT", Tasks.STYLIZED_IMAGE_CAPTIONING: "IMTEXT", Tasks.VISUALLY_GROUNDED_REASONING: "IMTEXT", - Tasks.HOAX_NEWS_CLASSIFICATION: "TEXT", - Tasks.CONCEPT_ALIGNMENT_CLASSIFICATION: "PAIRS", - Tasks.SPAN_BASED_ABSA: "SEQ_LABEL", Tasks.VIDEO_CAPTIONING: "VIDTEXT", Tasks.VIDEO_TO_TEXT_RETRIEVAL: "VIDTEXT", - Tasks.CONCEPT_ALIGNMENT_CLASSIFICATION: "PAIRS", Tasks.FACT_CHECKING: None, }