Skip MobileNetV1ModelTest::test_batching_equivalence for now (#35614)

fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-01-10 18:32:36 +01:00
committed by GitHub
parent 04eae987f3
commit 1e3c6c1f7d

View File

@@ -214,6 +214,8 @@ class MobileNetV1ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC
model = MobileNetV1Model.from_pretrained(model_name) model = MobileNetV1Model.from_pretrained(model_name)
self.assertIsNotNone(model) 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") @is_flaky(description="is_flaky https://github.com/huggingface/transformers/pull/31258")
def test_batching_equivalence(self): def test_batching_equivalence(self):
super().test_batching_equivalence() super().test_batching_equivalence()