From 1e3c6c1f7d7fbb4862ff036a9bdf96e6ae2e2511 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 10 Jan 2025 18:32:36 +0100 Subject: [PATCH] Skip `MobileNetV1ModelTest::test_batching_equivalence` for now (#35614) fix Co-authored-by: ydshieh --- tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py b/tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py index a04cfba45f..31153173d2 100644 --- a/tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py +++ b/tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py @@ -214,6 +214,8 @@ class MobileNetV1ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC model = MobileNetV1Model.from_pretrained(model_name) self.assertIsNotNone(model) + # TODO: ydshieh + @unittest.skip("skip for now as #35564 fails this test more frequently for this model") @is_flaky(description="is_flaky https://github.com/huggingface/transformers/pull/31258") def test_batching_equivalence(self): super().test_batching_equivalence()