formating

Author: qued

test_unstructured_inference/inference/test_layout_element.py

@@ -36,9 +36,7 @@ def test_layoutelement_inheritance_works_correctly():
 
     # Check that both properties are preserved
     assert element.source == Source.YOLOX, "LayoutElement should inherit source from TextRegion"
-    assert (
-        element.is_extracted
-    ), "LayoutElement should inherit is_extracted from TextRegion"
+    assert element.is_extracted, "LayoutElement should inherit is_extracted from TextRegion"
 
     # Check that to_dict() works correctly
     d = element.to_dict()

test_unstructured_inference/test_elements.py

@@ -495,9 +495,7 @@ def test_textregions_from_list_collects_sources():
     from unstructured_inference.inference.elements import TextRegion
 
     regions = [
-        TextRegion.from_coords(
-            0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True
-        ),
+        TextRegion.from_coords(0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True),
         TextRegion.from_coords(
             10,
             10,
@@ -531,9 +529,7 @@ def test_textregions_iter_elements_preserves_source():
     from unstructured_inference.inference.elements import TextRegion
 
     regions = [
-        TextRegion.from_coords(
-            0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True
-        ),
+        TextRegion.from_coords(0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True),
     ]
     text_regions = TextRegions.from_list(regions)
 
@@ -548,9 +544,7 @@ def test_textregions_slice_preserves_sources():
     from unstructured_inference.inference.elements import TextRegion
 
     regions = [
-        TextRegion.from_coords(
-            0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True
-        ),
+        TextRegion.from_coords(0, 0, 10, 10, text="first", source=Source.YOLOX, is_extracted=True),
         TextRegion.from_coords(
             10,
             10,

unstructured_inference/inference/layoutelement.py

@@ -257,7 +257,7 @@ def from_coords(
         y2: Union[int, float],
         text: Optional[str] = None,
         source: Optional[Source] = None,
-        is_extracted: bool = None,
+        is_extracted: Optional[bool] = None,
         type: Optional[str] = None,
         prob: Optional[float] = None,
         text_as_html: Optional[str] = None,