Skip to content

Commit 57ca083

Browse files
committed
remove extra parameterize for dtype
1 parent 8ed3b26 commit 57ca083

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5661,10 +5661,9 @@ def test_correctness_image(self, mean, std, dtype, fn):
56615661

56625662
@pytest.mark.skipif(not CVCUDA_AVAILABLE, reason="test requires CVCUDA")
56635663
@pytest.mark.parametrize(("mean", "std"), MEANS_STDS)
5664-
@pytest.mark.parametrize("dtype", [torch.float32])
56655664
@pytest.mark.parametrize("fn", [F.normalize, transform_cls_to_functional(transforms.Normalize)])
5666-
def test_correctness_cvcuda(self, mean, std, dtype, fn):
5667-
image = make_image(batch_dims=(1,), dtype=dtype, device="cuda")
5665+
def test_correctness_cvcuda(self, mean, std, fn):
5666+
image = make_image(batch_dims=(1,), dtype=torch.float32, device="cuda")
56685667
cvc_image = F.to_cvcuda_tensor(image)
56695668
actual = F._misc._normalize_cvcuda(cvc_image, mean=mean, std=std)
56705669
expected = fn(image, mean=mean, std=std)

0 commit comments

Comments
 (0)