Skip to content

Commit ccd4c1d

Browse files
committed
resolve tests failing due to cvcuda.Tensor
1 parent d9e3f83 commit ccd4c1d

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
@@ -3959,12 +3959,14 @@ def test_functional(self, make_input):
39593959
(F.gaussian_blur_video, tv_tensors.Video),
39603960
pytest.param(
39613961
F._misc._gaussian_blur_cvcuda,
3962-
cvcuda.Tensor,
3962+
"cvcuda.Tensor",
39633963
marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="CVCUDA is not available"),
39643964
),
39653965
],
39663966
)
39673967
def test_functional_signature(self, kernel, input_type):
3968+
if input_type == "cvcuda.Tensor":
3969+
input_type = _import_cvcuda().Tensor
39683970
check_functional_kernel_signature_match(F.gaussian_blur, kernel=kernel, input_type=input_type)
39693971

39703972
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)