type hint the invert tensor dict

justincdavis · justincdavis · commit 437ea99ba940 · 2025-12-04T13:51:37.000-08:00
diff --git a/torchvision/transforms/v2/functional/_color.py b/torchvision/transforms/v2/functional/_color.py
@@ -690,7 +690,7 @@ def invert_video(video: torch.Tensor) -> torch.Tensor:
     return invert_image(video)
 
 
-_invert_cvcuda_tensors = {}
+_invert_cvcuda_tensors: dict[str, "cvcuda.Tensor"] = {}
 
 
 def _invert_image_cvcuda(image: "cvcuda.Tensor") -> "cvcuda.Tensor":
