Ignore import-not-found

Author: AntoineSimoulin

torchvision/__init__.py

@@ -101,8 +101,8 @@ def _is_tracing():
 
 def _is_cvcuda_available() -> bool:
     try:
-        import cvcuda
-        import nvcv
+        import cvcuda  # type: ignore[import-not-found]
+        import nvcv  # type: ignore[import-not-found]
     except ImportError:
         return False
     return True

torchvision/transforms/v2/functional/_cvcuda.py

@@ -1,7 +1,8 @@
 import warnings
 
 import numpy as np
-import nvcv
+import nvcv  # type: ignore[import-not-found]
+import cvcuda  # type: ignore[import-not-found]
 import torch
 from torchvision.utils import _log_api_usage_once
 
@@ -200,8 +201,6 @@ def to_nvcv_tensor(pic, format=None):
         img_tensor = img_tensor.unsqueeze(0)  # H W -> N H W
 
     # Convert to NVCV tensor with NHWC layout
-    import cvcuda
-
     return cvcuda.as_tensor(img_tensor.contiguous(), nvcv.TensorLayout.NHWC)