test: update test to pass in TextRegions instead of list

Unstructured-IO · Oct 16, 2024 · 87278a9 · 87278a9
1 parent 963611c
commit 87278a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_unstructured/partition/pdf_image/test_inference_utils.py b/test_unstructured/partition/pdf_image/test_inference_utils.py
@@ -1,4 +1,4 @@
-from unstructured_inference.inference.elements import TextRegion
+from unstructured_inference.inference.elements import TextRegion, TextRegions
 from unstructured_inference.inference.layoutelement import LayoutElement
 
 from unstructured.documents.elements import ElementType
@@ -17,7 +17,7 @@ def test_merge_text_regions(mock_embedded_text_regions):
         text="LayoutParser: A Unified Toolkit for Deep Learning Based Document Image",
     )
 
-    merged_text_region = merge_text_regions(mock_embedded_text_regions)
+    merged_text_region = merge_text_regions(TextRegions.from_list(mock_embedded_text_regions))
     assert merged_text_region == expected