fix maybe

yuming-long · yuming-long · commit 8c08c2048915 · 2024-10-24T20:52:23.000-07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,6 @@
+## 0.8.1
+* fix: fix a bug caused by making LayoutElements subject from empty page
+
 ## 0.8.0
 
 * fix: fix missing source after cleaning layout elements
diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py
@@ -1 +1 @@
-__version__ = "0.8.0"  # pragma: no cover
+__version__ = "0.8.1"  # pragma: no cover
diff --git a/unstructured_inference/inference/layoutelement.py b/unstructured_inference/inference/layoutelement.py
@@ -145,7 +145,7 @@ def from_list(cls, elements: list):
             element_probs=np.array(class_probs),
             element_class_ids=class_ids,
             element_class_id_map=dict(zip(range(len(unique_ids)), unique_ids)),
-            source=elements[0].source,
+            source=elements[0].source if len_ele else None,
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ def from_list(cls, elements: list):`
`145`	`145`	`element_probs=np.array(class_probs),`
`146`	`146`	`element_class_ids=class_ids,`
`147`	`147`	`element_class_id_map=dict(zip(range(len(unique_ids)), unique_ids)),`
`148`		`- source=elements[0].source,`
	`148`	`+ source=elements[0].source if len_ele else None,`
`149`	`149`	`)`
`150`	`150`
`151`	`151`