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 19
19
import numpy as np
20
20
21
21
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
23
23
from transformers .utils import is_torch_available , is_torchvision_available , is_vision_available
24
24
25
25
from ...test_image_processing_common import ImageProcessingTestMixin , prepare_video_inputs
@@ -349,6 +349,17 @@ def test_call_pytorch(self):
349
349
350
350
@require_vision
351
351
@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
+ )
352
363
def test_slow_fast_equivalence_batched (self ):
353
364
if not self .test_slow_image_processor or not self .test_fast_image_processor :
354
365
self .skipTest (reason = "Skipping slow/fast equivalence test" )
You can’t perform that action at this time.
0 commit comments