mark test_save_load_fast_init_from_base as is_flaky (#20200)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-11-14 18:51:33 +01:00
committed by GitHub
parent af1a7c8ca3
commit 536e60d2c7

View File

@@ -49,6 +49,7 @@ from transformers.testing_utils import (
USER,
CaptureLogger,
TestCasePlus,
is_flaky,
is_pt_flax_cross_test,
is_pt_tf_cross_test,
is_staging_test,
@@ -340,6 +341,7 @@ class ModelTesterMixin:
if hasattr(module, "bias") and module.bias is not None:
module.bias.data.fill_(3)
@is_flaky()
def test_save_load_fast_init_from_base(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
base_class = MODEL_MAPPING[config.__class__]