Skip to content

Commit 8aadef5

Browse files
vfdev-5datumbox
andauthored
Try to fix lint/mypy failed CI (#5598)
Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 8f61f4c commit 8aadef5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/prototype/transforms/functional/_geometry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,9 +318,9 @@ def resized_crop_image_pil(
318318

319319
def _parse_five_crop_size(size: List[int]) -> List[int]:
320320
if isinstance(size, numbers.Number):
321-
size = (int(size), int(size))
321+
size = [int(size), int(size)]
322322
elif isinstance(size, (tuple, list)) and len(size) == 1:
323-
size = (size[0], size[0]) # type: ignore[assignment]
323+
size = [size[0], size[0]]
324324

325325
if len(size) != 2:
326326
raise ValueError("Please provide only two dimensions (h, w) for size.")

0 commit comments

Comments
 (0)