formatting

Author: qued

test_unstructured/partition/pdf_image/test_merge_elements.py

@@ -1,10 +1,13 @@
 from PIL import Image
-
+from unstructured_inference.constants import IsExtracted
+from unstructured_inference.inference.elements import Rectangle
 from unstructured_inference.inference.layout import DocumentLayout, PageLayout
 from unstructured_inference.inference.layoutelement import LayoutElement, LayoutElements
-from unstructured_inference.inference.elements import Rectangle
-from unstructured.partition.pdf_image.pdfminer_processing import merge_inferred_with_extracted_layout
-from unstructured_inference.constants import IsExtracted
+
+from unstructured.partition.pdf_image.pdfminer_processing import (
+    merge_inferred_with_extracted_layout,
+)
+
 
 def test_text_source_preserved_during_merge():
     """Test that text_source property is preserved when elements are merged."""

test_unstructured/partition/pdf_image/test_pdfminer_processing.py

@@ -4,7 +4,8 @@
 import pytest
 from pdfminer.layout import LAParams
 from PIL import Image
-from unstructured_inference.constants import Source as InferenceSource, IsExtracted
+from unstructured_inference.constants import IsExtracted
+from unstructured_inference.constants import Source as InferenceSource
 from unstructured_inference.inference.elements import (
     EmbeddedTextRegion,
     Rectangle,

unstructured/partition/pdf_image/pdfminer_processing.py

@@ -128,7 +128,9 @@ def _merge_extracted_into_inferred_when_almost_the_same(
     extracted_to_remove = extracted_layout.slice(extracted_almost_the_same_as_inferred)
     # copy here in case we change the extracted layout later
     inferred_layout.texts[inferred_indices_to_update] = extracted_to_remove.texts.copy()
-    inferred_layout.is_extracted_array[inferred_indices_to_update] = extracted_to_remove.is_extracted_array.copy()
+    inferred_layout.is_extracted_array[inferred_indices_to_update] = (
+        extracted_to_remove.is_extracted_array.copy()
+    )
     # use coords that can bound BOTH the inferred and extracted region as final bounding box coords
     inferred_layout.element_coords[inferred_indices_to_update] = _minimum_containing_coords(
         inferred_layout.slice(inferred_indices_to_update),