From a1d99499d490cfc83a44c857192f0d6911486eff Mon Sep 17 00:00:00 2001 From: bigcat88 Date: Sat, 19 Apr 2025 10:50:29 +0300 Subject: [PATCH] fixed typo in typing Signed-off-by: bigcat88 --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils.py b/utils.py index 7219b07..7890658 100644 --- a/utils.py +++ b/utils.py @@ -20,7 +20,7 @@ def temporary_env_var(key: str, new_value: str | None): del os.environ[key] -def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image]: +def images_to_pillow(images: Tensor | list[Tensor]) -> list[Image.Image]: pillow_images = [] for _bn, image in enumerate(images): i = 255.0 * image.cpu().numpy()