handle some comments from other prs review

Author: justincdavis

test/test_transforms_v2.py

@@ -3473,12 +3473,14 @@ def test_functional(self, make_input):
             (F.crop_keypoints, tv_tensors.KeyPoints),
             pytest.param(
                 F._geometry._crop_cvcuda,
-                _import_cvcuda().Tensor,
+                "cvcuda.Tensor",
                 marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA"),
             ),
         ],
     )
     def test_functional_signature(self, kernel, input_type):
+        if input_type == "cvcuda.Tensor":
+            input_type = _import_cvcuda().Tensor
         check_functional_kernel_signature_match(F.crop, kernel=kernel, input_type=input_type)
 
     @pytest.mark.parametrize("kwargs", CORRECTNESS_CROP_KWARGS)
@@ -4932,12 +4934,14 @@ def test_functional(self, make_input):
             (F.center_crop_keypoints, tv_tensors.KeyPoints),
             pytest.param(
                 F._geometry._center_crop_cvcuda,
-                _import_cvcuda().Tensor,
+                "cvcuda.Tensor",
                 marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA"),
             ),
         ],
     )
     def test_functional_signature(self, kernel, input_type):
+        if input_type == "cvcuda.Tensor":
+            input_type = _import_cvcuda().Tensor
         check_functional_kernel_signature_match(F.center_crop, kernel=kernel, input_type=input_type)
 
     @pytest.mark.parametrize(

torchvision/transforms/v2/functional/_geometry.py

@@ -1952,7 +1952,7 @@ def _crop_cvcuda(
 
 
 if CVCUDA_AVAILABLE:
-    _crop_cvcuda_registered = _register_kernel_internal(crop, _import_cvcuda().Tensor)(_crop_cvcuda)
+    _register_kernel_internal(crop, _import_cvcuda().Tensor)(_crop_cvcuda)
 
 
 def perspective(
@@ -2741,9 +2741,7 @@ def _center_crop_cvcuda(
 
 
 if CVCUDA_AVAILABLE:
-    _center_crop_cvcuda_registered = _register_kernel_internal(center_crop, _import_cvcuda().Tensor)(
-        _center_crop_cvcuda
-    )
+    _register_kernel_internal(center_crop, _import_cvcuda().Tensor)(_center_crop_cvcuda)
 
 
 def resized_crop(