Skip to content

GitLab

Unverified Commit e30bbb26 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Tests: tag `test_save_load_fast_init_from_base` as flaky (#28930)

parent 1709886e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/test_modeling_common.py
View file @ e30bbb26
...@@ -64,6 +64,7 @@ from transformers.models.auto.modeling_auto import ( ...@@ -64,6 +64,7 @@ from transformers.models.auto.modeling_auto import (
) )
from transformers.testing_utils import ( from transformers.testing_utils import (
CaptureLogger, CaptureLogger,
is_flaky,
is_pt_flax_cross_test, is_pt_flax_cross_test,
is_pt_tf_cross_test, is_pt_tf_cross_test,
require_accelerate, require_accelerate,
...@@ -381,6 +382,7 @@ class ModelTesterMixin: ...@@ -381,6 +382,7 @@ class ModelTesterMixin:
m.gradient_checkpointing, f"Module {n} does not have gradient_checkpointing set to False" 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): def test_save_load_fast_init_from_base(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
if config.__class__ not in MODEL_MAPPING: if config.__class__ not in MODEL_MAPPING:
......
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