Skip to content

Commit 7f869c7

Browse files
fix linting
1 parent 59ab66b commit 7f869c7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,11 +52,11 @@
5252
from torchvision.transforms.v2 import functional as F
5353
from torchvision.transforms.v2._utils import check_type, is_pure_tensor
5454
from torchvision.transforms.v2.functional._geometry import _get_perspective_coeffs, _parallelogram_to_bounding_boxes
55+
from torchvision.transforms.v2.functional._type_conversion import _import_cvcuda_modules
5556
from torchvision.transforms.v2.functional._utils import _get_kernel, _register_kernel_internal
5657

5758
try:
58-
import cvcuda # type: ignore[import-not-found]
59-
59+
_import_cvcuda_modules()
6060
CVCUDA_AVAILABLE = True
6161
except ImportError:
6262
CVCUDA_AVAILABLE = False

0 commit comments

Comments
 (0)