From d24bce5d9cd4a4d0ebb03b36acd99d3d1cb12994 Mon Sep 17 00:00:00 2001 From: j-t-1 <120829237+j-t-1@users.noreply.github.com> Date: Sat, 21 Dec 2024 11:31:42 +0000 Subject: [PATCH] STY: Use generator expressions (#3016) --- pypdf/_xobj_image_helpers.py | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/pypdf/_xobj_image_helpers.py b/pypdf/_xobj_image_helpers.py index 5687b3233..93ab1a8f5 100644 --- a/pypdf/_xobj_image_helpers.py +++ b/pypdf/_xobj_image_helpers.py @@ -152,7 +152,7 @@ def _extended_image_frombytes( if data_length % nb_pix != 0: raise exc k = nb_pix * len(mode) / data_length - data = b"".join([bytes((x,) * int(k)) for x in data]) + data = b"".join(bytes((x,) * int(k)) for x in data) img = Image.frombytes(mode, size, data) return img @@ -226,15 +226,11 @@ def _handle_flate( lookup = lookup[:expected_count] colors_arr = [lookup[:nb], lookup[nb:]] arr = b"".join( - [ - b"".join( - [ - colors_arr[1 if img.getpixel((x, y)) > 127 else 0] - for x in range(img.size[0]) - ] - ) - for y in range(img.size[1]) - ] + b"".join( + colors_arr[1 if img.getpixel((x, y)) > 127 else 0] + for x in range(img.size[0]) + ) + for y in range(img.size[1]) ) img = Image.frombytes(mode, img.size, arr) else: