From 83dfe9d2943e7837c9a6384b4f6e62de57c4735c Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 5 Sep 2025 12:23:07 +0200 Subject: [PATCH 1/2] fix --- tests/models/vitmatte/test_image_processing_vitmatte.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/vitmatte/test_image_processing_vitmatte.py b/tests/models/vitmatte/test_image_processing_vitmatte.py index 7201139702fc..d2a51833d147 100644 --- a/tests/models/vitmatte/test_image_processing_vitmatte.py +++ b/tests/models/vitmatte/test_image_processing_vitmatte.py @@ -267,7 +267,7 @@ def test_image_processor_preprocess_arguments(self): self.assertGreaterEqual(len(raised_warnings), 1) self.assertIn("extra_argument", messages) - @is_flaky() + @unittest.skip(reason="TODO: Yoni") 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") From 138ba90fc963d0fbf6979e9068b58fa93b867ca9 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 5 Sep 2025 12:25:16 +0200 Subject: [PATCH 2/2] fix --- tests/models/vitmatte/test_image_processing_vitmatte.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/models/vitmatte/test_image_processing_vitmatte.py b/tests/models/vitmatte/test_image_processing_vitmatte.py index d2a51833d147..644a90d03d31 100644 --- a/tests/models/vitmatte/test_image_processing_vitmatte.py +++ b/tests/models/vitmatte/test_image_processing_vitmatte.py @@ -23,7 +23,6 @@ from transformers.image_utils import load_image from transformers.testing_utils import ( - is_flaky, require_torch, require_torch_accelerator, require_vision,