From 3b720ad9a5a3a854e2f11d43a61a056b5afd3b61 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 8 Dec 2023 14:54:32 +0100 Subject: [PATCH] mark `test_initialization` as flaky in 2 model tests (#27906) fix Co-authored-by: ydshieh --- tests/models/dinov2/test_modeling_dinov2.py | 5 +++++ tests/models/vitdet/test_modeling_vitdet.py | 6 +++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/models/dinov2/test_modeling_dinov2.py b/tests/models/dinov2/test_modeling_dinov2.py index 4586f68ac5..8e68165754 100644 --- a/tests/models/dinov2/test_modeling_dinov2.py +++ b/tests/models/dinov2/test_modeling_dinov2.py @@ -19,6 +19,7 @@ import unittest from transformers import Dinov2Config from transformers.testing_utils import ( + is_flaky, require_torch, require_vision, slow, @@ -230,6 +231,10 @@ class Dinov2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): self.model_tester = Dinov2ModelTester(self) self.config_tester = ConfigTester(self, config_class=Dinov2Config, has_text_modality=False, hidden_size=37) + @is_flaky(max_attempts=3, description="`torch.nn.init.trunc_normal_` is flaky.") + def test_initialization(self): + super().test_initialization() + def test_config(self): self.config_tester.run_common_tests() diff --git a/tests/models/vitdet/test_modeling_vitdet.py b/tests/models/vitdet/test_modeling_vitdet.py index 45ee0acee3..2df1b79257 100644 --- a/tests/models/vitdet/test_modeling_vitdet.py +++ b/tests/models/vitdet/test_modeling_vitdet.py @@ -18,7 +18,7 @@ import unittest from transformers import VitDetConfig -from transformers.testing_utils import require_torch, torch_device +from transformers.testing_utils import is_flaky, require_torch, torch_device from transformers.utils import is_torch_available from ...test_backbone_common import BackboneTesterMixin @@ -175,6 +175,10 @@ class VitDetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): self.model_tester = VitDetModelTester(self) self.config_tester = ConfigTester(self, config_class=VitDetConfig, has_text_modality=False, hidden_size=37) + @is_flaky(max_attempts=3, description="`torch.nn.init.trunc_normal_` is flaky.") + def test_initialization(self): + super().test_initialization() + # TODO: Fix me (once this model gets more usage) @unittest.skip("Does not work on the tiny model as we keep hitting edge cases.") def test_cpu_offload(self):