Skip to content

Commit

Permalink
formats type fix
Browse files Browse the repository at this point in the history
  • Loading branch information
mwawrzos committed Jul 12, 2024
1 parent c4e7c35 commit 5673a51
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from enum import Enum, auto
from io import BytesIO
from pathlib import Path
from typing import Optional, Tuple, cast
from typing import List, Optional, Tuple, cast

import numpy as np
from genai_perf.exceptions import GenAIPerfException
Expand All @@ -15,7 +15,7 @@ class ImageFormat(Enum):


class RandomFormatBase64Encoder:
def __init__(self, image_formats: ImageFormat = ImageFormat.PNG):
def __init__(self, image_formats: List[ImageFormat] = [ImageFormat.PNG]):
self.image_formats = image_formats

def __call__(self, image):
Expand Down

0 comments on commit 5673a51

Please sign in to comment.