diff --git a/examples/seq2seq/rouge_cli.py b/examples/seq2seq/rouge_cli.py index b193581bc82..6a54a72eb58 100644 --- a/examples/seq2seq/rouge_cli.py +++ b/examples/seq2seq/rouge_cli.py @@ -9,7 +9,7 @@ def calculate_rouge_path(pred_path, tgt_path, save_path=None, **kwargs): tgt_lns = [x.strip() for x in open(tgt_path).readlines()][: len(pred_lns)] metrics = calculate_rouge(pred_lns, tgt_lns, **kwargs) if save_path is not None: - save_json(metrics, save_path) + save_json(metrics, save_path, indent=None) return metrics # these print nicely diff --git a/examples/seq2seq/run_eval.py b/examples/seq2seq/run_eval.py index 09830b66db7..910d430bddb 100755 --- a/examples/seq2seq/run_eval.py +++ b/examples/seq2seq/run_eval.py @@ -152,8 +152,7 @@ def run_generate(verbose=True): print(scores) if args.score_path is not None: - path = args.score_path - json.dump(scores, open(path, "w")) + json.dump(scores, open(args.score_path, "w")) return scores