diff --git a/test/transforms_v2_kernel_infos.py b/test/transforms_v2_kernel_infos.py index 3c3611cb8cc..a29e0429d1a 100644 --- a/test/transforms_v2_kernel_infos.py +++ b/test/transforms_v2_kernel_infos.py @@ -1925,9 +1925,6 @@ def sample_inputs_adjust_contrast_video(): yield ArgsKwargs(video_loader, contrast_factor=_ADJUST_CONTRAST_FACTORS[0]) -# TODO: this is just temporary to make CI green for release. We should add proper tolerances after -skip_adjust_contrast_jit = TestMark(("TestKernels", "test_scripted_vs_eager"), pytest.mark.skip(reason="Test is flaky")) - KERNEL_INFOS.extend( [ KernelInfo( @@ -1942,13 +1939,11 @@ def sample_inputs_adjust_contrast_video(): **float32_vs_uint8_pixel_difference(2), **cuda_vs_cpu_pixel_difference(), }, - test_marks=[skip_adjust_contrast_jit], ), KernelInfo( F.adjust_contrast_video, sample_inputs_fn=sample_inputs_adjust_contrast_video, closeness_kwargs=cuda_vs_cpu_pixel_difference(), - test_marks=[skip_adjust_contrast_jit], ), ] )