diff --git a/torchvision/prototype/transforms/functional/_geometry.py b/torchvision/prototype/transforms/functional/_geometry.py index 6c9309749af..84d1fe963c9 100644 --- a/torchvision/prototype/transforms/functional/_geometry.py +++ b/torchvision/prototype/transforms/functional/_geometry.py @@ -318,9 +318,9 @@ def resized_crop_image_pil( def _parse_five_crop_size(size: List[int]) -> List[int]: if isinstance(size, numbers.Number): - size = (int(size), int(size)) + size = [int(size), int(size)] elif isinstance(size, (tuple, list)) and len(size) == 1: - size = (size[0], size[0]) # type: ignore[assignment] + size = [size[0], size[0]] if len(size) != 2: raise ValueError("Please provide only two dimensions (h, w) for size.")