Skip to content

Commit

Permalink
remove unused imports
Browse files Browse the repository at this point in the history
  • Loading branch information
mwawrzos committed Jul 15, 2024
1 parent e0b43fd commit 1ef4f71
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 7 deletions.
Original file line number Diff line number Diff line change
@@ -1,11 +1,9 @@
import base64
from enum import Enum, auto
from io import BytesIO
from pathlib import Path
from typing import Generator, Optional, Tuple, cast
from typing import Optional, cast

import numpy as np
from genai_perf.exceptions import GenAIPerfException
from PIL import Image


Expand All @@ -29,7 +27,7 @@ def __init__(
self._image_width_stddev = image_width_stddev
self._image_height_stddev = image_height_stddev
self.image_format = image_format
self.rng = rng or np.random.default_rng()
self.rng = cast(np.random.Generator, rng or np.random.default_rng())

def __iter__(self):
return self
Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,8 @@
import base64
from io import BytesIO
from pathlib import Path
from unittest.mock import patch

import numpy as np
import pytest
from genai_perf.exceptions import GenAIPerfException
from genai_perf.llm_inputs.synthetic_image_generator import (
ImageFormat,
SyntheticImageGenerator,
Expand Down

0 comments on commit 1ef4f71

Please sign in to comment.