update transformed_types

justincdavis · justincdavis · commit 23816f4b33d1 · 2025-12-04T14:21:32.000-08:00
diff --git a/torchvision/transforms/v2/_geometry.py b/torchvision/transforms/v2/_geometry.py
@@ -686,6 +686,8 @@ class RandomAffine(Transform):
 
     _v1_transform_cls = _transforms.RandomAffine
 
+    _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
+
     def __init__(
         self,
         degrees: Union[numbers.Number, Sequence],
