Skip to content

Commit f0b51c4

Browse files
committed
resolve tests failing due to cvcuda.Tensor
1 parent aef48fd commit f0b51c4

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
@@ -3906,12 +3906,14 @@ def test_functional(self, make_input):
39063906
(F.gaussian_blur_video, tv_tensors.Video),
39073907
pytest.param(
39083908
F._misc._gaussian_blur_cvcuda,
3909-
cvcuda.Tensor,
3909+
"cvcuda.Tensor",
39103910
marks=pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="CVCUDA is not available"),
39113911
),
39123912
],
39133913
)
39143914
def test_functional_signature(self, kernel, input_type):
3915+
if input_type == "cvcuda.Tensor":
3916+
input_type = _import_cvcuda().Tensor
39153917
check_functional_kernel_signature_match(F.gaussian_blur, kernel=kernel, input_type=input_type)
39163918

39173919
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)