Update run_squad.py

Indentation change so that the output "nbest_predictions.json" is not empty.
2019-05-22 17:27:59 +08:00
parent 3fc63f126d
commit 9e7bc51b95
1 changed files with 1 additions and 1 deletions
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -617,7 +617,7 @@ def write_predictions(all_examples, all_features, all_results, n_best_size,
                all_predictions[example.qas_id] = ""
            else:
                all_predictions[example.qas_id] = best_non_null_entry.text
-            all_nbest_json[example.qas_id] = nbest_json
+        all_nbest_json[example.qas_id] = nbest_json

    with open(output_prediction_file, "w") as writer:
        writer.write(json.dumps(all_predictions, indent=4) + "\n")