Skip to content

Commit 9e87618

Browse files
Fix docstring and type hint for resize (#27104)
fix docstring and type hint for resize
1 parent ef23b68 commit 9e87618

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/image_transforms.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def get_resize_output_image_size(
274274

275275

276276
def resize(
277-
image,
277+
image: np.ndarray,
278278
size: Tuple[int, int],
279279
resample: "PILImageResampling" = None,
280280
reducing_gap: Optional[int] = None,
@@ -286,7 +286,7 @@ def resize(
286286
Resizes `image` to `(height, width)` specified by `size` using the PIL library.
287287
288288
Args:
289-
image (`PIL.Image.Image` or `np.ndarray` or `torch.Tensor`):
289+
image (`np.ndarray`):
290290
The image to resize.
291291
size (`Tuple[int, int]`):
292292
The size to use for resizing the image.

0 commit comments

Comments
 (0)