From ef367187265648cae4f9134786d4b5d2146c06af Mon Sep 17 00:00:00 2001 From: Ryan Gibson Date: Fri, 16 Aug 2024 17:14:51 -0400 Subject: [PATCH] Remove extraneous typing exclusions --- stego_lsb/LSBSteg.py | 4 ++-- tests/test_lsbsteg.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/stego_lsb/LSBSteg.py b/stego_lsb/LSBSteg.py index 0c79920..1f3528e 100755 --- a/stego_lsb/LSBSteg.py +++ b/stego_lsb/LSBSteg.py @@ -93,8 +93,8 @@ def hide_message_in_image(input_image: Image.Image, message: Union[str, bytes], log.debug(f"{f'{message_size} bytes hidden':<30} in {time() - start:.2f}s") start = time() - # PIL expects a sequence of tuples, one per pixel TODO: this expression is too complicated for typing to handle? - input_image.putdata(cast(List[int], list(zip(*[iter(flattened_color_data)] * num_channels)))) # type: ignore + # PIL expects a sequence of tuples, one per pixel + input_image.putdata(cast(List[int], list(zip(*[iter(flattened_color_data)] * num_channels)))) log.debug(f"{'Image overwritten':<30} in {time() - start:.2f}s") return input_image diff --git a/tests/test_lsbsteg.py b/tests/test_lsbsteg.py index 04eb404..395ed06 100644 --- a/tests/test_lsbsteg.py +++ b/tests/test_lsbsteg.py @@ -14,8 +14,7 @@ class TestLSBSteg(unittest.TestCase): def write_random_image(self, filename: str, width: int, height: int, num_channels: int) -> None: image_data = np.random.randint(0, 256, size=(height, width, num_channels), dtype=np.uint8) - # TODO: Image.fromarray() appears to have buggy typing? - with Image.fromarray(image_data) as image: # type: ignore + with Image.fromarray(image_data) as image: image.save(filename) def write_random_file(self, filename: str, num_bytes: int) -> None: