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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions

View File

@ -41,8 +41,7 @@ FLAKY_TEST_FAILURE_PATTERNS = [
"ConnectionError", # Connection transient error
"FileNotFoundError", # Raised by `datasets` on Hub failures
"PIL.UnidentifiedImageError", # Raised by `PIL.Image.open` on connection issues
"HTTPError.*502", # Hub-related
"HTTPError.*504", # Hub-related
"HTTPError", # Also catches HfHubHTTPError
"AssertionError: Tensor-likes are not close!", # `torch.testing.assert_close`, we might have unlucky random values
# TODO: error downloading tokenizer's `merged.txt` from hub can cause all the exceptions below. Throw and handle
# them under a single message.

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")