Fix KeyError when evaluating

remove the debug code to fix KeyError when evaluating
cooelf · Jun 27, 2020 · db4aa56 · db4aa56 · yangcheng2020 · Jun 27, 2020
1 parent 18c68c7
commit db4aa56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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