From 5b57024c84b76b9c655f7ee795787d22c5f51024 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sun, 25 Jun 2023 08:42:06 -0700 Subject: [PATCH] cleanup --stats output --- scripts/benchmark.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/scripts/benchmark.py b/scripts/benchmark.py index 9ea32f686..b3ce8f4bb 100644 --- a/scripts/benchmark.py +++ b/scripts/benchmark.py @@ -134,9 +134,10 @@ def main(): ) all_results = run_test_threaded.gather(tqdm=True) - print() - print() - print() + if not args.stats_only: + print() + print() + print() summarize_results(all_results, total_tests) @@ -186,7 +187,8 @@ def summarize_results(all_results, total_tests=None): def run_test(testdir, model_name, edit_format, retries, no_test, verbose, stats_only): - dump(testdir) + if not stats_only: + dump(testdir) if not os.path.isdir(testdir): print("Not a dir:", testdir)