bbox float to int

bouncia · Oct 13, 2023 · b307ac4 · b307ac4
1 parent bf77b0b
commit b307ac4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/bisheng-langchain/bisheng_langchain/document_loaders/elem_unstrcutured_loader.py b/src/bisheng-langchain/bisheng_langchain/document_loaders/elem_unstrcutured_loader.py
@@ -52,7 +52,8 @@ def merge_partitions(partitions):
                 doc_content.append(text_elem_sep + text)
 
         last_label = label
-        metadata['bboxes'].extend(extra_data['bboxes'])
+        metadata['bboxes'].extend(
+            list(map(lambda x: list(map(int, x)), extra_data['bboxes'])))
         metadata['pages'].extend(extra_data['pages'])
         metadata['types'].extend(extra_data['types'])