diff --git a/transformer-mrc/transformers/data/processors/squad.py b/transformer-mrc/transformers/data/processors/squad.py index 70516b6..3d3ba90 100644 --- a/transformer-mrc/transformers/data/processors/squad.py +++ b/transformer-mrc/transformers/data/processors/squad.py @@ -527,9 +527,9 @@ def _create_examples(self, input_data, set_type): examples = [] for entry in tqdm(input_data): title = entry["title"] - for paragraph in entry["paragraphs"][:1]: + for paragraph in entry["paragraphs"]: context_text = paragraph["context"] - for qa in paragraph["qas"][:1]: + for qa in paragraph["qas"]: qas_id = qa["id"] question_text = qa["question"] start_position_character = None