File tree 1 file changed +2
-2
lines changed
torchvision/prototype/transforms/functional 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -2089,10 +2089,10 @@ def ten_crop(
2089
2089
return ten_crop_image_tensor (inpt , size , vertical_flip = vertical_flip )
2090
2090
elif isinstance (inpt , datapoints .Image ):
2091
2091
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 ]
2093
2093
elif isinstance (inpt , datapoints .Video ):
2094
2094
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 ]
2096
2096
elif isinstance (inpt , PIL .Image .Image ):
2097
2097
return ten_crop_image_pil (inpt , size , vertical_flip = vertical_flip )
2098
2098
else :
You can’t perform that action at this time.
0 commit comments