Skip to content

Commit d010e82

Browse files
authored
fix ten_crop datapoint return (#7260)
1 parent 30bb1ce commit d010e82

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
@@ -2089,10 +2089,10 @@ def ten_crop(
20892089
return ten_crop_image_tensor(inpt, size, vertical_flip=vertical_flip)
20902090
elif isinstance(inpt, datapoints.Image):
20912091
output = ten_crop_image_tensor(inpt.as_subclass(torch.Tensor), size, vertical_flip=vertical_flip)
2092-
return [datapoints.Image.wrap_like(inpt, item) for item in output]
2092+
return tuple(datapoints.Image.wrap_like(inpt, item) for item in output) # type: ignore[return-value]
20932093
elif isinstance(inpt, datapoints.Video):
20942094
output = ten_crop_video(inpt.as_subclass(torch.Tensor), size, vertical_flip=vertical_flip)
2095-
return [datapoints.Video.wrap_like(inpt, item) for item in output]
2095+
return tuple(datapoints.Video.wrap_like(inpt, item) for item in output) # type: ignore[return-value]
20962096
elif isinstance(inpt, PIL.Image.Image):
20972097
return ten_crop_image_pil(inpt, size, vertical_flip=vertical_flip)
20982098
else:

0 commit comments

Comments
 (0)