Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
generate_configs.py 2.81 KB
Newer Older
lintangsutawika's avatar
add blimp  
lintangsutawika committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95 

import yaml

all_subtasks = [
    'adjunct_island',
    'anaphor_gender_agreement',
    'anaphor_number_agreement',
    'animate_subject_passive',
    'animate_subject_trans',
    'causative',
    'complex_NP_island',
    'coordinate_structure_constraint_complex_left_branch',
    'coordinate_structure_constraint_object_extraction',
    'determiner_noun_agreement_1',
    'determiner_noun_agreement_2',
    'determiner_noun_agreement_irregular_1',
    'determiner_noun_agreement_irregular_2',
    'determiner_noun_agreement_with_adj_2',
    'determiner_noun_agreement_with_adj_irregular_1',
    'determiner_noun_agreement_with_adj_irregular_2',
    'determiner_noun_agreement_with_adjective_1',
    'distractor_agreement_relational_noun',
    'distractor_agreement_relative_clause',
    'drop_argument',
    'ellipsis_n_bar_1',
    'ellipsis_n_bar_2',
    'existential_there_object_raising',
    'existential_there_quantifiers_1',
    'existential_there_quantifiers_2',
    'existential_there_subject_raising',
    'expletive_it_object_raising',
    'inchoative',
    'intransitive',
    'irregular_past_participle_adjectives',
    'irregular_past_participle_verbs',
    'irregular_plural_subject_verb_agreement_1',
    'irregular_plural_subject_verb_agreement_2',
    'left_branch_island_echo_question',
    'left_branch_island_simple_question',
    'matrix_question_npi_licensor_present',
    'npi_present_1',
    'npi_present_2',
    'only_npi_licensor_present',
    'only_npi_scope',
    'passive_1',
    'passive_2',
    'principle_A_c_command',
    'principle_A_case_1',
    'principle_A_case_2',
    'principle_A_domain_1',
    'principle_A_domain_2',
    'principle_A_domain_3',
    'principle_A_reconstruction',
    'regular_plural_subject_verb_agreement_1',
    'regular_plural_subject_verb_agreement_2',
    'sentential_negation_npi_licensor_present',
    'sentential_negation_npi_scope',
    'sentential_subject_island',
    'superlative_quantifiers_1',
    'superlative_quantifiers_2',
    'tough_vs_raising_1',
    'tough_vs_raising_2',
    'transitive',
    'wh_island',
    'wh_questions_object_gap',
    'wh_questions_subject_gap',
    'wh_questions_subject_gap_long_distance',
    'wh_vs_that_no_gap',
    'wh_vs_that_no_gap_long_distance',
    'wh_vs_that_with_gap',
    'wh_vs_that_with_gap_long_distance'
]


def main() -> None:
    for task in all_subtasks:

        file_name = f"{task}.yaml"
        try:
            with open(f"{file_name}", "w") as f:
                f.write("# Generated by utils.py\n")
                yaml.dump(
                    {
                        "include": "template_yaml",
                        "task": "blimp_"+task,
                        "dataset_name": task,
                    },
                    f,
                )
        except FileExistsError:
            pass


if __name__ == "__main__":
    main()