Skip to content

Commit 2c68fc3

Browse files
committed
add adjust saturation
1 parent 61b237c commit 2c68fc3

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
@@ -6335,10 +6335,7 @@ def test_correctness_image(self, make_input, color_space, saturation_factor):
63356335
actual = F.adjust_saturation(image, saturation_factor=saturation_factor)
63366336

63376337
if make_input is make_image_cvcuda:
6338-
actual = F.cvcuda_to_tensor(actual).to(device="cpu")
6339-
actual = actual.squeeze(0)
6340-
image = F.cvcuda_to_tensor(image)
6341-
image = image.squeeze(0)
6338+
image = cvcuda_to_pil_compatible_tensor(image)
63426339

63436340
expected = F.to_image(F.adjust_saturation(F.to_pil_image(image), saturation_factor=saturation_factor))
63446341

0 commit comments

Comments
 (0)