Skip to content

Commit 7a231b1

Browse files
committed
further simplify todtype test
1 parent 736a2e6 commit 7a231b1

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

test/test_transforms_v2.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2753,12 +2753,10 @@ def test_image_correctness(self, input_dtype, output_dtype, device, scale, make_
27532753
out = F.to_dtype(input, dtype=output_dtype, scale=scale)
27542754

27552755
if make_input == make_image_cvcuda:
2756-
expected = self.reference_convert_dtype_image_tensor(
2757-
F.cvcuda_to_tensor(input), dtype=output_dtype, scale=scale
2758-
)
2756+
input = F.cvcuda_to_tensor(input)
27592757
out = F.cvcuda_to_tensor(out)
2760-
else:
2761-
expected = self.reference_convert_dtype_image_tensor(input, dtype=output_dtype, scale=scale)
2758+
2759+
expected = self.reference_convert_dtype_image_tensor(input, dtype=output_dtype, scale=scale)
27622760

27632761
atol = self._get_dtype_conversion_atol(input_dtype, output_dtype, scale)
27642762
torch.testing.assert_close(out, expected, rtol=0, atol=atol)

0 commit comments

Comments
 (0)