From 9011318431c7f4093eb12f1a7b4a0fb88ce1685d Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Fri, 11 Mar 2022 18:02:18 +0000 Subject: [PATCH] Try to fix lint/mypy failed CI --- torchvision/prototype/transforms/functional/_geometry.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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.")