fix: cover no source case

Unstructured-IO · Oct 18, 2024 · 9c36d2a · 9c36d2a
1 parent c052187
commit 9c36d2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):