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

Fix form and key value area groups

Browse Source 
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-12-16 15:01:27 +01:00
parent efc25225ac
commit 0fd50e53be
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docling/utils/glm_utils.py
View File

@ -263,9 +263,9 @@ def to_docling_document(doc_glm, update_name_label=False) -> DoclingDocument:
label = DocItemLabel(ptype) label = DocItemLabel(ptype)
group_label = GroupLabel.UNSPECIFIED group_label = GroupLabel.UNSPECIFIED
if label == DocItemLabel.FORM: if label == DocItemLabel.FORM:
group_label == GroupLabel.FORM_AREA group_label = GroupLabel.FORM_AREA
elif label == DocItemLabel.KEY_VALUE_REGION: elif label == DocItemLabel.KEY_VALUE_REGION:
group_label == GroupLabel.KEY_VALUE_AREA group_label = GroupLabel.KEY_VALUE_AREA
container_el = doc.add_group(label=group_label) container_el = doc.add_group(label=group_label)