diff --git a/unstructured_inference/inference/layoutelement.py b/unstructured_inference/inference/layoutelement.py index 42d1c3c1..332df603 100644 --- a/unstructured_inference/inference/layoutelement.py +++ b/unstructured_inference/inference/layoutelement.py @@ -90,7 +90,7 @@ def concatenate(cls, groups: Iterable[LayoutElements]) -> LayoutElements: element_probs=np.concatenate(probs), element_class_ids=np.concatenate(class_ids), element_class_id_map=class_id_map, - source=sources[0], + source=sources[0] if sources else None, ) def as_list(self):