Solve mypy warning.

This commit is contained in:
Cédric Bonhomme 2017-05-04 13:25:14 +02:00
parent 13f0fb8086
commit a279b21a70
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D

View file

@ -82,11 +82,9 @@ def hide(input_image_file: str,
# Save the new pixel # Save the new pixel
if img.mode == 'RGBA': if img.mode == 'RGBA':
updated_pixel = (r, g, b, pixel[3]) encoded.putpixel((col, row), (r, g, b, pixel[3]))
else: else:
updated_pixel = (r, g, b) encoded.putpixel((col, row), (r, g, b))
encoded.putpixel((col, row), updated_pixel)
index += 3 index += 3
else: else: