def serialize()

in src/huggingface_inference_toolkit/serialization/image_utils.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def serialize(image, accept=None):
        if isinstance(image, Image.Image):
            img_byte_arr = BytesIO()
            image.save(img_byte_arr, format=accept.split("/")[-1].upper())
            img_byte_arr = img_byte_arr.getvalue()
            return img_byte_arr
        else:
            raise ValueError(f"Can only serialize PIL.Image.Image, got {type(image)}")