From 495729427045c7a58e040fa9bf6df81c16f54208 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 7 Jul 2023 11:55:21 +0200 Subject: [PATCH] Fix flaky `test_for_warning_if_padding_and_no_attention_mask` (#24706) fix Co-authored-by: ydshieh --- tests/models/bert/test_modeling_bert.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/models/bert/test_modeling_bert.py b/tests/models/bert/test_modeling_bert.py index db02174071..52c8035d8a 100644 --- a/tests/models/bert/test_modeling_bert.py +++ b/tests/models/bert/test_modeling_bert.py @@ -584,6 +584,9 @@ class BertModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin # Check for warnings if the attention_mask is missing. logger = logging.get_logger("transformers.modeling_utils") + # clear cache so we can test the warning is emitted (from `warning_once`). + logger.warning_once.cache_clear() + with CaptureLogger(logger) as cl: model = BertModel(config=config) model.to(torch_device)