From ef2277ec4e64ccd19f6a2a092850d4503882ff7a Mon Sep 17 00:00:00 2001 From: John <43506685+Coniferish@users.noreply.github.com> Date: Thu, 12 Sep 2024 22:13:49 -0400 Subject: [PATCH] linting fix 2 --- unstructured_inference/inference/layout.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unstructured_inference/inference/layout.py b/unstructured_inference/inference/layout.py index 944f2548..f6c0a8cc 100644 --- a/unstructured_inference/inference/layout.py +++ b/unstructured_inference/inference/layout.py @@ -98,7 +98,7 @@ def from_image_file( try: image = Image.open(filename) format = image.format - images = [] + images: list[Image.Image]= [] for i, im in enumerate(ImageSequence.Iterator(image)): im = im.convert("RGB") im.format = format @@ -143,7 +143,7 @@ def __init__( image_metadata = {} self.image_metadata = image_metadata self.image_path = image_path - self.image_array: Union[np.ndarray, None] = None + self.image_array: Union[np.ndarray[Any, Any], None] = None self.document_filename = document_filename self.number = number self.detection_model = detection_model @@ -198,7 +198,7 @@ def get_elements_with_detection_model( return inferred_layout - def _get_image_array(self) -> Union[np.ndarray, None]: + def _get_image_array(self) -> Union[np.ndarray[Any, Any], None]: """Converts the raw image into a numpy array.""" if self.image_array is None: if self.image: