Skip to content

Commit 429f77f

Browse files
committed
fix tests crashing before run without cvcuda
1 parent 79ea0da commit 429f77f

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
@@ -5592,12 +5592,14 @@ def test_functional(self, make_input):
55925592
(F.normalize_video, tv_tensors.Video),
55935593
pytest.param(
55945594
F._misc._normalize_cvcuda,
5595-
_import_cvcuda().Tensor,
5595+
"cvcuda.Tensor",
55965596
marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA"),
55975597
),
55985598
],
55995599
)
56005600
def test_functional_signature(self, kernel, input_type):
5601+
if input_type == "cvcuda.Tensor":
5602+
input_type = _import_cvcuda().Tensor
56015603
check_functional_kernel_signature_match(F.normalize, kernel=kernel, input_type=input_type)
56025604

56035605
def test_functional_error(self):

0 commit comments

Comments
 (0)