Tests: tag test_save_load_fast_init_from_base as flaky (#28930)

huggingface · May 14, 2024 · 14cd2d7 · 14cd2d7
1 parent ce771ab
commit 14cd2d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -64,6 +64,7 @@
 )
 from transformers.testing_utils import (
     CaptureLogger,
+    is_flaky,
     is_pt_flax_cross_test,
     is_pt_tf_cross_test,
     require_accelerate,
@@ -381,6 +382,7 @@ def test_gradient_checkpointing_enable_disable(self):
                         m.gradient_checkpointing, f"Module {n} does not have gradient_checkpointing set to False"
                     )
 
+    @is_flaky(description="low likelihood of failure, reason not yet discovered")
     def test_save_load_fast_init_from_base(self):
         config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
         if config.__class__ not in MODEL_MAPPING: