Skip to content

Commit efef692

Browse files
committed
add adjust saturation
1 parent bd5c9db commit efef692

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

test/test_transforms_v2.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6283,10 +6283,7 @@ def test_correctness_image(self, make_input, color_space, saturation_factor):
62836283
actual = F.adjust_saturation(image, saturation_factor=saturation_factor)
62846284

62856285
if make_input is make_image_cvcuda:
6286-
actual = F.cvcuda_to_tensor(actual).to(device="cpu")
6287-
actual = actual.squeeze(0)
6288-
image = F.cvcuda_to_tensor(image)
6289-
image = image.squeeze(0)
6286+
image = cvcuda_to_pil_compatible_tensor(image)
62906287

62916288
expected = F.to_image(F.adjust_saturation(F.to_pil_image(image), saturation_factor=saturation_factor))
62926289

0 commit comments

Comments
 (0)