SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Mark MobileNetV1ModelTest::test_batching_equivalence as flaky (#31258)

Browse Source 
* Mark MobileNetV1ModelTest::test_batching_equivalence as flaky

* Add link to issue

* woops
This commit is contained in:
amyeroberts
2024-06-06 14:47:58 +01:00
committed by GitHub
parent 681183974a
commit c53fcd8381
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/mobilenet_v1/test_modeling_mobilenet_v1.py
View File

@@ -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
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():