Unverified Commit cd332463 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

skip flaky adjust_saturation kernel tests (#7348)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent 08887344
......@@ -2064,6 +2064,9 @@ def sample_inputs_adjust_saturation_video():
yield ArgsKwargs(video_loader, saturation_factor=_ADJUST_SATURATION_FACTORS[0])
# TODO: this is just temporary to make CI green for release. We should add proper tolerances after
skip_adjust_saturation_cuda = TestMark(("TestKernels", "test_cuda_vs_cpu"), pytest.mark.skip(reason="Test is flaky"))
KERNEL_INFOS.extend(
[
KernelInfo(
......@@ -2077,10 +2080,12 @@ KERNEL_INFOS.extend(
**pil_reference_pixel_difference(),
**float32_vs_uint8_pixel_difference(2),
},
test_marks=[skip_adjust_saturation_cuda],
),
KernelInfo(
F.adjust_saturation_video,
sample_inputs_fn=sample_inputs_adjust_saturation_video,
test_marks=[skip_adjust_saturation_cuda],
),
]
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment