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

fix for KeyError in tableformer prediction

Browse Source 
Signed-off-by: Maksym Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maksym Lysak 2025-01-31 16:11:58 +01:00
parent 70d68b6164
commit 8f79e95e74
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
docling/models/table_structure_model.py
View File

@ -210,11 +210,22 @@ class TableStructureModel(BasePageModel):
table_cells.append(tc)
# Retrieving cols/rows, after post processing:
num_rows = table_out["predict_details"]["num_rows"]
num_cols = table_out["predict_details"]["num_cols"]
otsl_seq = table_out["predict_details"]["prediction"][
"rs_seq"
]
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"]
tbl = Table(
otsl_seq=otsl_seq,