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
if img.mode == 'RGBA':
updated_pixel = (r, g, b, pixel[3])
encoded.putpixel((col, row), (r, g, b, pixel[3]))
else:
updated_pixel = (r, g, b)
encoded.putpixel((col, row), updated_pixel)
encoded.putpixel((col, row), (r, g, b))
index += 3
else: