From 7c323e1af0124768f1f675b591187429c689bff8 Mon Sep 17 00:00:00 2001 From: Mike Gerber Date: Thu, 25 May 2023 18:36:46 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20dinglehopper-summarize:=20Handle?= =?UTF-8?q?=20reports=20without=20difference=20stats?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dinglehopper/cli_summarize.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/dinglehopper/cli_summarize.py b/dinglehopper/cli_summarize.py index 1cf1c91..f1c0ca1 100644 --- a/dinglehopper/cli_summarize.py +++ b/dinglehopper/cli_summarize.py @@ -33,10 +33,13 @@ def process(reports_folder, occurrences_threshold=1): cer_sum += cer wer_sum += wer - for key, value in report_data["differences"]["character_level"].items(): - diff_c[key] = diff_c.get(key, 0) + value - for key, value in report_data["differences"]["word_level"].items(): - diff_w[key] = diff_w.get(key, 0) + value + try: + for key, value in report_data["differences"]["character_level"].items(): + diff_c[key] = diff_c.get(key, 0) + value + for key, value in report_data["differences"]["word_level"].items(): + diff_w[key] = diff_w.get(key, 0) + value + except KeyError: + pass if len(cer_list) == 0: click.echo(f"No reports found in folder '{os.path.abspath(reports_folder)}'")