mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-30 14:04:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

removed comments

Browse Source 
Signed-off-by: Maksym Lysak <mly@zurich.ibm.com>
This commit is contained in:
Maksym Lysak 2025-03-28 12:19:45 +01:00
parent 787c6d8ace
commit a4f6762aef
1 changed files with 2 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
docling/models/table_structure_model.py
View File

@ -227,22 +227,10 @@ class TableStructureModel(BasePageModel):
if sp is not None: if sp is not None:
tcells = sp.get_cells_in_bbox( tcells = sp.get_cells_in_bbox(
cell_unit=TextCellUnit.WORD, cell_unit=TextCellUnit.WORD,
bbox=table_cluster.bbox.to_bottom_left_origin( bbox=table_cluster.bbox,
page.size.height
),
) )
# Transform origin of returned cells:
tcells2 = []
for tcell in tcells:
# Do the copy to not affect cells that are in two (or more) table regions
tcell = tcell.model_copy(deep=True)
tcell.rect = tcell.rect.to_top_left_origin(
page.size.height
)
tcells2.append(tcell)
tcells = tcells2
else: else:
print("Otherwise - we use normal (line/phrase) cells") # Otherwise - we use normal (line/phrase) cells
tcells = table_cluster.cells tcells = table_cluster.cells
tokens = [] tokens = []
for c in tcells: for c in tcells:
@ -268,7 +256,6 @@ class TableStructureModel(BasePageModel):
) )
table_out = tf_output[0] table_out = tf_output[0]
table_cells = [] table_cells = []
# print("len(tf_responses)={}".format(len(table_out["tf_responses"])))
for element in table_out["tf_responses"]: for element in table_out["tf_responses"]:
if not self.do_cell_matching: if not self.do_cell_matching: