SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

replace print with logger (#12368)

Browse Source
This commit is contained in:
Bhadresh Savani
2021-06-26 17:31:25 +01:00
committed by GitHub
parent 9a7545943d
commit ff5cdc086b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/pytorch/question-answering/utils_qa.py
View File

@@ -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")