diff --git a/image_prediction/image_extractor/extractors/parsable.py b/image_prediction/image_extractor/extractors/parsable.py index 7d26b29..894c360 100644 --- a/image_prediction/image_extractor/extractors/parsable.py +++ b/image_prediction/image_extractor/extractors/parsable.py @@ -193,7 +193,7 @@ def has_alpha_channel(doc: Doc, xref: int): maybe_image = load_image_handle_from_xref(doc, xref) maybe_smask = maybe_image["smask"] if maybe_image else None - if maybe_smask: # Use monad + if maybe_smask: # TODO: Use monad. return any([doc.extract_image(maybe_smask) is not None, bool(fitz.Pixmap(doc, maybe_smask).alpha)]) else: try: @@ -261,7 +261,7 @@ def normalize_channels(array: np.ndarray): @lru_cache(maxsize=None) -def load_image_handle_from_xref(doc: Doc, xref: int) -> Union[dict, None]: # TODO: use Monad +def load_image_handle_from_xref(doc: Doc, xref: int) -> Union[dict, None]: # TODO: Use monad. return doc.extract_image(xref)