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

Tests: tag test_save_load_fast_init_from_base as flaky (#28930)

Browse Source
This commit is contained in:
Joao Gante
2024-02-12 14:43:34 +00:00
committed by GitHub
parent 1709886eba
commit e30bbb2685
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

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