diff --git a/dan/ocr/evaluate.py b/dan/ocr/evaluate.py index 041d309e6e9be7919360f8bc29dc721ce404068e..c5e79fcee4e3797b065a1519fde00db99f642551 100644 --- a/dan/ocr/evaluate.py +++ b/dan/ocr/evaluate.py @@ -27,16 +27,16 @@ logger = logging.getLogger(__name__) NERVAL_THRESHOLD = 0.30 -def parse_threshold(arg): +def parse_threshold(value): try: - f = float(arg) + value = float(value) except ValueError: raise ArgumentTypeError("Must be a floating point number.") - if f < 0 or f > 1: + if value < 0 or value > 1: raise ArgumentTypeError("Must be between 0 and 1.") - return f + return value def add_evaluate_parser(subcommands) -> None: diff --git a/dan/ocr/manager/metrics.py b/dan/ocr/manager/metrics.py index d7b3f6f233a9995773b399cd44e063154cd918ec..4e668a68c3e6d58477747fd40355637e7c71506f 100644 --- a/dan/ocr/manager/metrics.py +++ b/dan/ocr/manager/metrics.py @@ -162,7 +162,6 @@ class MetricManager: metrics["time"] = [values["time"]] gt, prediction = values["str_y"], values["str_x"] - for metric_name in metric_names: match metric_name: case (