Skip to content

Commit 6821135

Browse files
committed
fix tests crashing before run without cvcuda
1 parent dbf4a5c commit 6821135

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/test_transforms_v2.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5539,12 +5539,14 @@ def test_functional(self, make_input):
55395539
(F.normalize_video, tv_tensors.Video),
55405540
pytest.param(
55415541
F._misc._normalize_cvcuda,
5542-
_import_cvcuda().Tensor,
5542+
"cvcuda.Tensor",
55435543
marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA"),
55445544
),
55455545
],
55465546
)
55475547
def test_functional_signature(self, kernel, input_type):
5548+
if input_type == "cvcuda.Tensor":
5549+
input_type = _import_cvcuda().Tensor
55485550
check_functional_kernel_signature_match(F.normalize, kernel=kernel, input_type=input_type)
55495551

55505552
def test_functional_error(self):

0 commit comments

Comments
 (0)