Skip to content

Commit 086e294

Browse files
committed
Enable all qa tasks in dataset prep
1 parent 37ba397 commit 086e294

File tree

1 file changed

+19
-19
lines changed

1 file changed

+19
-19
lines changed

prep_datasets.py

+19-19
Original file line numberDiff line numberDiff line change
@@ -128,25 +128,25 @@ def main():
128128

129129
filenames = [
130130
'qa1_single-supporting-fact',
131-
# 'qa2_two-supporting-facts',
132-
# 'qa3_three-supporting-facts',
133-
# 'qa4_two-arg-relations',
134-
# 'qa5_three-arg-relations',
135-
# 'qa6_yes-no-questions',
136-
# 'qa7_counting',
137-
# 'qa8_lists-sets',
138-
# 'qa9_simple-negation',
139-
# 'qa10_indefinite-knowledge',
140-
# 'qa11_basic-coreference',
141-
# 'qa12_conjunction',
142-
# 'qa13_compound-coreference',
143-
# 'qa14_time-reasoning',
144-
# 'qa15_basic-deduction',
145-
# 'qa16_basic-induction',
146-
# 'qa17_positional-reasoning',
147-
# 'qa18_size-reasoning',
148-
# 'qa19_path-finding',
149-
# 'qa20_agents-motivations',
131+
'qa2_two-supporting-facts',
132+
'qa3_three-supporting-facts',
133+
'qa4_two-arg-relations',
134+
'qa5_three-arg-relations',
135+
'qa6_yes-no-questions',
136+
'qa7_counting',
137+
'qa8_lists-sets',
138+
'qa9_simple-negation',
139+
'qa10_indefinite-knowledge',
140+
'qa11_basic-coreference',
141+
'qa12_conjunction',
142+
'qa13_compound-coreference',
143+
'qa14_time-reasoning',
144+
'qa15_basic-deduction',
145+
'qa16_basic-induction',
146+
'qa17_positional-reasoning',
147+
'qa18_size-reasoning',
148+
'qa19_path-finding',
149+
'qa20_agents-motivations',
150150
]
151151

152152
tar = tarfile.open(os.path.join(SRC_DIR, 'babi_tasks_data_1_20_v1.2.tar.gz'))

0 commit comments

Comments
 (0)