Make the flaky list a little more general (#36704)

* Make the flaky list a little more general

* Trigger tests

* Make the flaky list a little more general
This commit is contained in:
Matt
2025-03-14 12:15:32 +00:00
committed by GitHub
parent 53742b11f5
commit 72861e11eb
2 changed files with 3 additions and 2 deletions

View File

@@ -29,6 +29,7 @@ from transformers import AutoImageProcessor, BatchFeature
from transformers.image_utils import AnnotationFormat, AnnotionFormat
from transformers.testing_utils import (
check_json_file_has_correct_format,
is_flaky,
require_torch,
require_torch_gpu,
require_vision,
@@ -212,6 +213,7 @@ class ImageProcessingTestMixin:
@require_vision
@require_torch
@is_flaky()
def test_fast_is_faster_than_slow(self):
if not self.test_slow_image_processor or not self.test_fast_image_processor:
self.skipTest(reason="Skipping speed test")