chore: Refactor remove function to handle unsupported input types

This commit is contained in:
Daniel Gatis 2024-08-04 09:55:19 -03:00
parent 90536b6858
commit 688b34841b

View File

@ -241,7 +241,7 @@ def remove(
""" """
if isinstance(data, bytes) or force_return_bytes: if isinstance(data, bytes) or force_return_bytes:
return_type = ReturnType.BYTES return_type = ReturnType.BYTES
img = Image.open(io.BytesIO(data)) img = Image.open(io.BytesIO(cast(bytes, data)))
elif isinstance(data, PILImage): elif isinstance(data, PILImage):
return_type = ReturnType.PILLOW return_type = ReturnType.PILLOW
img = data img = data
@ -249,7 +249,11 @@ def remove(
return_type = ReturnType.NDARRAY return_type = ReturnType.NDARRAY
img = Image.fromarray(data) img = Image.fromarray(data)
else: else:
raise ValueError("Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format(type(data))) raise ValueError(
"Input type {} is not supported. Try using force_return_bytes=True to force python bytes output".format(
type(data)
)
)
putalpha = kwargs.pop("putalpha", False) putalpha = kwargs.pop("putalpha", False)