diff --git a/image_prediction/image_extractor/extractors/parsable.py b/image_prediction/image_extractor/extractors/parsable.py index 9f684a3..1a8e1ef 100644 --- a/image_prediction/image_extractor/extractors/parsable.py +++ b/image_prediction/image_extractor/extractors/parsable.py @@ -107,11 +107,7 @@ def load_image_handle_from_xref(doc, xref): def has_alpha_channel(doc, xref): maybe_image = load_image_handle_from_xref(doc, xref) - if maybe_image: - maybe_smask = doc.extract_image(maybe_image["smask"]) - return maybe_smask is not None - else: - return False + return doc.extract_image(maybe_image["smask"]) is not None if maybe_image else False def xref_to_image(doc, xref):