Unverified Commit c53fcd83 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Mark MobileNetV1ModelTest::test_batching_equivalence as flaky (#31258)

* Mark MobileNetV1ModelTest::test_batching_equivalence as flaky

* Add link to issue

* woops
parent 68118397
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,7 @@
import unittest import unittest
from transformers import MobileNetV1Config from transformers import MobileNetV1Config
from transformers.testing_utils import require_torch, require_vision, slow, torch_device from transformers.testing_utils import is_flaky, require_torch, require_vision, slow, torch_device
from transformers.utils import cached_property, is_torch_available, is_vision_available from transformers.utils import cached_property, is_torch_available, is_vision_available
from ...test_configuration_common import ConfigTester from ...test_configuration_common import ConfigTester
...@@ -214,6 +214,10 @@ class MobileNetV1ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC ...@@ -214,6 +214,10 @@ class MobileNetV1ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC
model = MobileNetV1Model.from_pretrained(model_name) model = MobileNetV1Model.from_pretrained(model_name)
self.assertIsNotNone(model) self.assertIsNotNone(model)
@is_flaky(description="is_flaky https://github.com/huggingface/transformers/pull/31258")
def test_batching_equivalence(self):
super().test_batching_equivalence()
# We will verify our results on an image of cute cats # We will verify our results on an image of cute cats
def prepare_img(): def prepare_img():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment