Skip to content

Commit 5fbeac3

Browse files
committed
move transformed type check to Rotate transform
1 parent 550656f commit 5fbeac3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchvision/transforms/v2/_geometry.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
is_pure_tensor,
3030
query_size,
3131
)
32+
from .functional._utils import is_cvcuda_tensor
3233

3334
CVCUDA_AVAILABLE = _is_cvcuda_available()
3435

@@ -606,6 +607,8 @@ class RandomRotation(Transform):
606607

607608
_v1_transform_cls = _transforms.RandomRotation
608609

610+
_transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
611+
609612
def __init__(
610613
self,
611614
degrees: Union[numbers.Number, Sequence],

0 commit comments

Comments
 (0)