From 2e94bda7fae9a8d312d327da6076b0fa5c4cc844 Mon Sep 17 00:00:00 2001 From: Kadir Date: Sun, 25 Aug 2024 09:50:41 +0200 Subject: [PATCH] refac --- crawler/rec/floorplan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crawler/rec/floorplan.py b/crawler/rec/floorplan.py index b942c76..562ab08 100644 --- a/crawler/rec/floorplan.py +++ b/crawler/rec/floorplan.py @@ -17,9 +17,9 @@ def inference(image_path): return output, predictions -def extract_total_sqm(deplot_input_str): +def extract_total_sqm(input_str: str): sqmregex = r"(\d+\.\d*) ?(sq ?m|sq. ?m)" - matches = re.findall(sqmregex, deplot_input_str.lower()) + matches = re.findall(sqmregex, input_str.lower()) if len(matches) == 0: return None sqms = [float(m[0]) for m in matches]