fix: fix col line filter to handle empty list

This commit is contained in:
iriley 2024-04-26 14:28:44 +02:00
parent 12bb7ee25f
commit fee357872f
2 changed files with 6 additions and 3 deletions

View File

@ -138,6 +138,8 @@ FILTERS = {
def filter_fp_col_lines(line_list: list[int], filt_sums: Array) -> list[int]:
if not line_list:
return []
centers = list(
np.where(
(filt_sums[1:-1] < filt_sums[:-2]) * (filt_sums[1:-1] < filt_sums[2:])

View File

@ -18,6 +18,7 @@ def test_table_inference_smoke():
}
data = {"pdf": pdf_bytes, "vlp_output": vlp_mock}
output = list(pl(data))
lines = output[0]["tableLines"]
# assert len(lines) > 1
# assert all(map(lambda item: sorted(item.keys()) == ["x1", "x2", "y1", "y2"], lines))
if output:
lines = output[0]["tableLines"]
# assert len(lines) > 1
# assert all(map(lambda item: sorted(item.keys()) == ["x1", "x2", "y1", "y2"], lines))