Skip to content

Commit 15ff139

Browse files
authored
fix: deterministic page sorting in reading-order model (#141)
Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
1 parent a9d92a7 commit 15ff139

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docling_ibm_models/reading_order/reading_order_rb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def predict_reading_order(
112112
page_to_footers[page_no] = self._predict_page(elems)
113113

114114
sorted_elements = []
115-
for page_no in page_nos:
115+
for page_no in sorted(page_nos):
116116
sorted_elements.extend(page_to_headers[page_no])
117117
sorted_elements.extend(page_to_elems[page_no])
118118
sorted_elements.extend(page_to_footers[page_no])

0 commit comments

Comments
 (0)