From 546ae98dbaf73391d4eb906b582074d0155529cb Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Wed, 15 Feb 2023 20:05:21 +0100 Subject: [PATCH] fix ten_crop datapoint return --- 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 840223908ac..22731bb157f 100644 --- a/torchvision/prototype/transforms/functional/_geometry.py +++ b/torchvision/prototype/transforms/functional/_geometry.py @@ -2089,10 +2089,10 @@ def ten_crop( return ten_crop_image_tensor(inpt, size, vertical_flip=vertical_flip) elif isinstance(inpt, datapoints.Image): output = ten_crop_image_tensor(inpt.as_subclass(torch.Tensor), size, vertical_flip=vertical_flip) - return [datapoints.Image.wrap_like(inpt, item) for item in output] + return tuple(datapoints.Image.wrap_like(inpt, item) for item in output) # type: ignore[return-value] elif isinstance(inpt, datapoints.Video): output = ten_crop_video(inpt.as_subclass(torch.Tensor), size, vertical_flip=vertical_flip) - return [datapoints.Video.wrap_like(inpt, item) for item in output] + return tuple(datapoints.Video.wrap_like(inpt, item) for item in output) # type: ignore[return-value] elif isinstance(inpt, PIL.Image.Image): return ten_crop_image_pil(inpt, size, vertical_flip=vertical_flip) else: