From 32c61a70b3fe40186aba24dbeb4bc14f820bb0db Mon Sep 17 00:00:00 2001 From: Daniil Galtsev Date: Fri, 27 Oct 2023 05:34:16 +0000 Subject: [PATCH] fix docstring and type hint for resize --- src/transformers/image_transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/image_transforms.py b/src/transformers/image_transforms.py index e3d0d23474d3..e9cb93db651d 100644 --- a/src/transformers/image_transforms.py +++ b/src/transformers/image_transforms.py @@ -274,7 +274,7 @@ def get_resize_output_image_size( def resize( - image, + image: np.ndarray, size: Tuple[int, int], resample: "PILImageResampling" = None, reducing_gap: Optional[int] = None, @@ -286,7 +286,7 @@ def resize( Resizes `image` to `(height, width)` specified by `size` using the PIL library. Args: - image (`PIL.Image.Image` or `np.ndarray` or `torch.Tensor`): + image (`np.ndarray`): The image to resize. size (`Tuple[int, int]`): The size to use for resizing the image.