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: