From 5df3d1e85b283432184b529c2c083f29d7fb5a3e Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Tue, 19 Apr 2022 14:29:02 +0200 Subject: [PATCH] return -> yield from --- .../image_extractor/extractors/parsable.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/image_prediction/image_extractor/extractors/parsable.py b/image_prediction/image_extractor/extractors/parsable.py index 70ef701..9ca98c9 100644 --- a/image_prediction/image_extractor/extractors/parsable.py +++ b/image_prediction/image_extractor/extractors/parsable.py @@ -58,7 +58,7 @@ def extract_pages(doc, page_range): page_range = range(page_range.start + 1, page_range.stop + 1) pages = map(doc.load_page, page_range) - return pages + yield from pages @lru_cache(maxsize=None) @@ -67,7 +67,7 @@ def get_images_on_page(doc, page: fitz.Page): xrefs = map(itemgetter("xref"), image_infos) images = map(partial(xref_to_image, doc), xrefs) - return images + yield from images def get_metadata_for_images_on_page(doc, page: fitz.Page): @@ -118,7 +118,7 @@ def validate_coords_and_passthrough(metadata): def filter_out_tiny_images(metadata): - return filterfalse(tiny, metadata) + yield from filterfalse(tiny, metadata) def validate_size_and_passthrough(metadata): @@ -126,7 +126,7 @@ def validate_size_and_passthrough(metadata): def add_page_metadata(page, metadata): - return map(partial(merge, get_page_metadata(page)), metadata) + yield from map(partial(merge, get_page_metadata(page)), metadata) def add_alpha_channel_info(doc, page, metadata): @@ -139,7 +139,7 @@ def add_alpha_channel_info(doc, page, metadata): metadata = starmap(merge, zip(page_to_alpha_mapping_per_image(page), metadata)) - return metadata + yield from metadata @lru_cache(maxsize=None)