File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change 1919import numpy as np
2020
2121from 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
2323from transformers .utils import is_torch_available , is_torchvision_available , is_vision_available
2424
2525from ...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" )
You can’t perform that action at this time.
0 commit comments