From b854312b08fd3ce1a9758aa5f06007da11e15a5b Mon Sep 17 00:00:00 2001 From: iriley Date: Mon, 13 May 2024 08:40:31 +0200 Subject: [PATCH] fix: check nonzero list length in filter_fp_col_lines --- pyproject.toml | 2 +- src/cv_analysis/table_inference.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 02ac964..3822a1c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "cv-analysis-service" -version = "2.15.1" +version = "2.16.1" description = "" authors = ["Isaac Riley "] readme = "README.md" diff --git a/src/cv_analysis/table_inference.py b/src/cv_analysis/table_inference.py index 62c3896..c341427 100644 --- a/src/cv_analysis/table_inference.py +++ b/src/cv_analysis/table_inference.py @@ -135,9 +135,11 @@ FILTERS: dict[str, dict[int, np.ndarray | None]] = { def filter_fp_col_lines(line_list: list[int], filt_sums: Array) -> list[int]: - if not line_list: + if not list(line_list): return [] centers = list(np.where((filt_sums[1:-1] < filt_sums[:-2]) * (filt_sums[1:-1] < filt_sums[2:]))[0] + 1) + if not centers: + return [] if line_list[0] > centers[0]: centers = centers[1:] + [len(filt_sums) - 1]