Aaron Mueller commited on
Commit
1bbb1d0
1 Parent(s): a5487ef

enumerate blimp tasks

Browse files
Files changed (1) hide show
  1. src/display/utils.py +18 -1
src/display/utils.py CHANGED
@@ -80,7 +80,24 @@ TEXT_TASKS = {
80
  "glue": ["cola", "sst2", "mrpc", "qqp", "mnli", "mnli-mm", "qnli", "rte",
81
  "boolq", "multirc", "wsc"],
82
  # Lots of BLiMP tasks – use verifier function below to see if you've included everything.
83
- "blimp": [taskname.split(".jsonl")[0] for taskname in os.listdir("evaluation_data/blimp_filtered/")],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
84
  "blimp_supplement": ["hypernym", "qa_congruence_easy", "qa_congruence_tricky",
85
  "subject_aux_inversion", "turn_taking"],
86
  "ewok": ["agent-properties", "material-dynamics", "material-properties", "physical-dynamics",
 
80
  "glue": ["cola", "sst2", "mrpc", "qqp", "mnli", "mnli-mm", "qnli", "rte",
81
  "boolq", "multirc", "wsc"],
82
  # Lots of BLiMP tasks – use verifier function below to see if you've included everything.
83
+ "blimp": ["adjunct_island","anaphor_gender_agreement","anaphor_number_agreement","animate_subject_passive","animate_subject_trans",
84
+ "causative","complex_NP_island","coordinate_structure_constraint_complex_left_branch","coordinate_structure_constraint_object_extraction","determiner_noun_agreement_1",
85
+ "determiner_noun_agreement_2","determiner_noun_agreement_irregular_1","determiner_noun_agreement_irregular_2","determiner_noun_agreement_with_adjective_1",
86
+ "determiner_noun_agreement_with_adj_2","determiner_noun_agreement_with_adj_irregular_1","determiner_noun_agreement_with_adj_irregular_2","distractor_agreement_relational_noun",
87
+ "distractor_agreement_relative_clause","drop_argument","ellipsis_n_bar_1","ellipsis_n_bar_2",
88
+ "existential_there_object_raising", "existential_there_quantifiers_1",
89
+ "existential_there_quantifiers_2", "existential_there_subject_raising", "expletive_it_object_raising",
90
+ "inchoative", "intransitive","irregular_past_participle_adjectives", "irregular_past_participle_verbs",
91
+ "irregular_plural_subject_verb_agreement_1", "irregular_plural_subject_verb_agreement_2", "left_branch_island_echo_question", "left_branch_island_simple_question",
92
+ "matrix_question_npi_licensor_present", "npi_present_1", "npi_present_2", "only_npi_licensor_present", "only_npi_scope", "passive_1", "passive_2",
93
+ "principle_A_case_1", "principle_A_case_2", "principle_A_c_command", "principle_A_domain_1",
94
+ "principle_A_domain_2", "principle_A_domain_3", "principle_A_reconstruction", "regular_plural_subject_verb_agreement_1",
95
+ "regular_plural_subject_verb_agreement_2", "sentential_negation_npi_licensor_present", "sentential_negation_npi_scope", "sentential_subject_island",
96
+ "superlative_quantifiers_1", "superlative_quantifiers_2", "tough_vs_raising_1", "tough_vs_raising_2",
97
+ "transitive", "wh_island", "wh_questions_object_gap", "wh_questions_subject_gap",
98
+ "wh_questions_subject_gap_long_distance", "wh_vs_that_no_gap", "wh_vs_that_no_gap_long_distance", "wh_vs_that_with_gap",
99
+ "wh_vs_that_with_gap_long_distance"
100
+ ],
101
  "blimp_supplement": ["hypernym", "qa_congruence_easy", "qa_congruence_tricky",
102
  "subject_aux_inversion", "turn_taking"],
103
  "ewok": ["agent-properties", "material-dynamics", "material-properties", "physical-dynamics",