Skip to content

Commit 7efb4c8

Browse files
authored
Flaky CI is annoying (#40543)
* mark flaky * and the non batch one
1 parent 828a27f commit 7efb4c8

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

tests/models/tvp/test_image_processing_tvp.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
import numpy as np
2020

2121
from transformers.image_transforms import PaddingMode
22-
from transformers.testing_utils import require_torch, require_vision
22+
from transformers.testing_utils import is_flaky, require_torch, require_vision
2323
from transformers.utils import is_torch_available, is_torchvision_available, is_vision_available
2424

2525
from ...test_image_processing_common import ImageProcessingTestMixin, prepare_video_inputs
@@ -349,6 +349,17 @@ def test_call_pytorch(self):
349349

350350
@require_vision
351351
@require_torch
352+
@is_flaky(
353+
description="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
354+
)
355+
def test_slow_fast_equivalence(self):
356+
super().test_slow_fast_equivalence()
357+
358+
@require_vision
359+
@require_torch
360+
@is_flaky(
361+
description="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
362+
)
352363
def test_slow_fast_equivalence_batched(self):
353364
if not self.test_slow_image_processor or not self.test_fast_image_processor:
354365
self.skipTest(reason="Skipping slow/fast equivalence test")

0 commit comments

Comments
 (0)