Skip to content

GitLab

Unverified Commit ff5cdc08 authored by Bhadresh Savani's avatar Bhadresh Savani Committed by GitHub
Browse files

replace print with logger (#12368)

parent 9a754594
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/pytorch/question-answering/utils_qa.py
View file @ ff5cdc08
...@@ -413,14 +413,14 @@ def postprocess_qa_predictions_with_beam_search( ...@@ -413,14 +413,14 @@ def postprocess_qa_predictions_with_beam_search(
output_dir, "null_odds.json" if prefix is None else f"{prefix}_null_odds.json" output_dir, "null_odds.json" if prefix is None else f"{prefix}_null_odds.json"
) )
print(f"Saving predictions to {prediction_file}.") logger.info(f"Saving predictions to {prediction_file}.")
with open(prediction_file, "w") as writer: with open(prediction_file, "w") as writer:
writer.write(json.dumps(all_predictions, indent=4) + "\n") writer.write(json.dumps(all_predictions, indent=4) + "\n")
print(f"Saving nbest_preds to {nbest_file}.") logger.info(f"Saving nbest_preds to {nbest_file}.")
with open(nbest_file, "w") as writer: with open(nbest_file, "w") as writer:
writer.write(json.dumps(all_nbest_json, indent=4) + "\n") writer.write(json.dumps(all_nbest_json, indent=4) + "\n")
if version_2_with_negative: if version_2_with_negative:
print(f"Saving null_odds to {null_odds_file}.") logger.info(f"Saving null_odds to {null_odds_file}.")
with open(null_odds_file, "w") as writer: with open(null_odds_file, "w") as writer:
writer.write(json.dumps(scores_diff_json, indent=4) + "\n") writer.write(json.dumps(scores_diff_json, indent=4) + "\n")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment