mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-08-01 23:12:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: rewrite cumbersome dictionary checking

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2025-01-31 16:26:27 +01:00
parent 8f79e95e74
commit bdeeb74534
1 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
docling/models/table_structure_model.py
View File

@ -209,23 +209,16 @@ class TableStructureModel(BasePageModel):
tc.bbox = tc.bbox.scaled(1 / self.scale)
table_cells.append(tc)
assert "predict_details" in table_out
# Retrieving cols/rows, after post processing:
num_rows = 0
num_cols = 0
otsl_seq = []
if "predict_details" in table_out:
if "num_rows" in table_out["predict_details"]:
num_rows = table_out["predict_details"]["num_rows"]
if "num_cols" in table_out["predict_details"]:
num_cols = table_out["predict_details"]["num_cols"]
if "prediction" in table_out["predict_details"]:
if (
"rs_seq"
in table_out["predict_details"]["prediction"]
):
otsl_seq = table_out["predict_details"][
"prediction"
]["rs_seq"]
num_rows = table_out["predict_details"].get("num_rows", 0)
num_cols = table_out["predict_details"].get("num_cols", 0)
otsl_seq = (
table_out["predict_details"]
.get("prediction", {})
.get("rs_seq", [])
)
tbl = Table(
otsl_seq=otsl_seq,