Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions tests/models/tvp/test_image_processing_tvp.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
import numpy as np

from transformers.image_transforms import PaddingMode
from transformers.testing_utils import is_flaky, require_torch, require_vision
from transformers.testing_utils import require_torch, require_vision
from transformers.utils import is_torch_available, is_torchvision_available, is_vision_available

from ...test_image_processing_common import ImageProcessingTestMixin, prepare_video_inputs
Expand Down Expand Up @@ -349,16 +349,16 @@ def test_call_pytorch(self):

@require_vision
@require_torch
@is_flaky(
description="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
@unittest.skip(
reason="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
)
def test_slow_fast_equivalence(self):
super().test_slow_fast_equivalence()

@require_vision
@require_torch
@is_flaky(
description="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
@unittest.skip(
reason="FIXME: @yoni probably because of an extra 'time' dimension and since image processors don't handle it well?"
)
def test_slow_fast_equivalence_batched(self):
if not self.test_slow_image_processor or not self.test_fast_image_processor:
Expand Down