diff --git a/image_prediction/image_extractor/extractors/parsable.py b/image_prediction/image_extractor/extractors/parsable.py index a76e071..9cd04e5 100644 --- a/image_prediction/image_extractor/extractors/parsable.py +++ b/image_prediction/image_extractor/extractors/parsable.py @@ -1,6 +1,6 @@ import io from functools import partial, lru_cache -from itertools import chain, starmap +from itertools import chain, starmap, filterfalse from operator import itemgetter, truth import fitz @@ -72,7 +72,7 @@ def get_metadata_for_images_on_page(page: fitz.Page): image_infos = get_image_infos(page) metadata = map(get_image_metadata, image_infos) metadata = validate_coords_and_passthrough(metadata) - metadata = filter(tiny, metadata) + metadata = filterfalse(tiny, metadata) metadata = validate_size_and_passthrough(metadata) metadata = map(partial(merge, get_page_metadata(page)), metadata) @@ -114,7 +114,7 @@ def get_image_metadata(image_info): def tiny(metadata): - return metadata[Info.WIDTH] * metadata[Info.HEIGHT] + return metadata[Info.WIDTH] * metadata[Info.HEIGHT] <= 4 def get_page_metadata(page):