chore(documentai-toolbox): fix linting and formatting

Author: chalmerlowe

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/__init__.py

@@ -25,7 +25,14 @@
 from .utilities import docai_utilities, gcs_utilities
 from .wrappers import document, entity, page
 
-__all__ = ("document", "page", "entity", "converter", "docai_utilities", "gcs_utilities")
+__all__ = (
+    "document",
+    "page",
+    "entity",
+    "converter",
+    "docai_utilities",
+    "gcs_utilities",
+)
 
 
 class Python37DeprecationWarning(DeprecationWarning):  # pragma: NO COVER

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/converters/config/bbox_conversion.py

@@ -190,7 +190,12 @@ def convert_bbox_to_docproto_bbox(block: Block) -> documentai.BoundingPoly:
     y_multiplier = 1.0
     normalized_vertices: List[documentai.NormalizedVertex] = []
 
-    if block.page_width and block.page_height and block.docproto_width is not None and block.docproto_height is not None:
+    if (
+        block.page_width
+        and block.page_height
+        and block.docproto_width is not None
+        and block.docproto_height is not None
+    ):
         x_multiplier = _get_multiplier(
             docproto_coordinate=block.docproto_width,
             external_coordinate=block.page_width,

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/converters/config/block.py

@@ -74,45 +74,19 @@ class Block:
     """
     type_: Any = dataclasses.field(init=True, repr=False)
     text: Any = dataclasses.field(init=True, repr=False)
-    bounding_box: Any = dataclasses.field(
-        init=True, repr=False, default=None
-    )
-    block_references: Any = dataclasses.field(
-        init=True, repr=False, default=None
-    )
-    block_id: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    confidence: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    page_number: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    page_width: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    page_height: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_width: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_height: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_type: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_unit: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_x: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
-    bounding_y: Any = dataclasses.field(
-        init=False, repr=False, default=None
-    )
+    bounding_box: Any = dataclasses.field(init=True, repr=False, default=None)
+    block_references: Any = dataclasses.field(init=True, repr=False, default=None)
+    block_id: Any = dataclasses.field(init=False, repr=False, default=None)
+    confidence: Any = dataclasses.field(init=False, repr=False, default=None)
+    page_number: Any = dataclasses.field(init=False, repr=False, default=None)
+    page_width: Any = dataclasses.field(init=False, repr=False, default=None)
+    page_height: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_width: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_height: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_type: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_unit: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_x: Any = dataclasses.field(init=False, repr=False, default=None)
+    bounding_y: Any = dataclasses.field(init=False, repr=False, default=None)
     docproto_width: Optional[float] = dataclasses.field(
         init=False, repr=False, default=None
     )
@@ -203,7 +177,9 @@ def load_blocks_from_schema(
             b = Block(
                 type_=block_type,
                 text=block_text,
-                bounding_box=_get_target_object(entity, normalized_vertices) if normalized_vertices is not None else None,
+                bounding_box=_get_target_object(entity, normalized_vertices)
+                if normalized_vertices is not None
+                else None,
             )
 
             if id_:

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/utilities/gcs_utilities.py

@@ -20,7 +20,11 @@
 
 from google.api_core.gapic_v1 import client_info
 
-from google.cloud import documentai, documentai_toolbox, storage  # type: ignore[attr-defined]
+from google.cloud import (
+    documentai,  # type: ignore[attr-defined]
+    documentai_toolbox,
+    storage,
+)
 from google.cloud.documentai_toolbox import constants
 
 

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/wrappers/entity.py

@@ -66,7 +66,7 @@ class Entity:
     def __post_init__(self, page_offset: Optional[int]) -> None:
         if page_offset is None:
             page_offset = 0
-            
+
         self.type_ = self.documentai_object.type_
 
         if self.documentai_object.mention_text:

packages/google-cloud-documentai-toolbox/google/cloud/documentai_toolbox/wrappers/page.py

@@ -20,14 +20,15 @@
 from functools import cached_property
 from typing import Iterable, List, Optional, Type, TypeVar
 
-T = TypeVar("T", bound="_BasePageElement")
 
 import pandas as pd
 
 from google.cloud import documentai
 from google.cloud.documentai_toolbox.constants import ElementWithLayout
 from google.cloud.documentai_toolbox.utilities import docai_utilities
 
+T = TypeVar("T", bound="_BasePageElement")
+
 
 @dataclasses.dataclass
 class Table:
@@ -182,9 +183,7 @@ def _text_segment(self) -> documentai.Document.TextAnchor.TextSegment:
         """
         return self.documentai_object.layout.text_anchor.text_segments[0]
 
-    def _get_children_of_element(
-        self, potential_children: List[T]
-    ) -> List[T]:
+    def _get_children_of_element(self, potential_children: List[T]) -> List[T]:
         """
         Filters potential child elements to identify only those fully contained within this element.