"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "8679bd714476244ae32657fb8ff63dbb2c087865"
Unverified Commit 536e60d2 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

mark `test_save_load_fast_init_from_base` as `is_flaky` (#20200)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent af1a7c8c
...@@ -49,6 +49,7 @@ from transformers.testing_utils import ( ...@@ -49,6 +49,7 @@ from transformers.testing_utils import (
USER, USER,
CaptureLogger, CaptureLogger,
TestCasePlus, TestCasePlus,
is_flaky,
is_pt_flax_cross_test, is_pt_flax_cross_test,
is_pt_tf_cross_test, is_pt_tf_cross_test,
is_staging_test, is_staging_test,
...@@ -340,6 +341,7 @@ class ModelTesterMixin: ...@@ -340,6 +341,7 @@ class ModelTesterMixin:
if hasattr(module, "bias") and module.bias is not None: if hasattr(module, "bias") and module.bias is not None:
module.bias.data.fill_(3) module.bias.data.fill_(3)
@is_flaky()
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()
base_class = MODEL_MAPPING[config.__class__] base_class = MODEL_MAPPING[config.__class__]
......
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