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

feat: Add option to control empty clusters in layout postprocessing (#1940)

Browse Source 
Add option to control empty clusters in layout postprocessing

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer
2025-07-14 18:32:01 +02:00
committed by GitHub
parent 95e70962f1
commit a436be7367
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docling/utils/layout_postprocessor.py
View File

@@ -267,8 +267,9 @@ class LayoutPostprocessor:
# Initial cell assignment
clusters = self._assign_cells_to_clusters(clusters)
# Remove clusters with no cells
clusters = [cluster for cluster in clusters if cluster.cells]
# Remove clusters with no cells (if keep_empty_clusters is False)
if not self.options.keep_empty_clusters:
clusters = [cluster for cluster in clusters if cluster.cells]
# Handle orphaned cells
unassigned = self._find_unassigned_cells(clusters)