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

Mark test_encoder_decoder_model_generate for vision_encoder_deocder as flaky (#28842)

Browse Source 
Mark test as flaky
This commit is contained in:
amyeroberts
2024-02-02 16:57:08 +00:00
committed by GitHub
parent 80d50076c8
commit 3d2900e829
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
View File

@@ -23,6 +23,7 @@ from packaging import version
from transformers import DonutProcessor, NougatProcessor, TrOCRProcessor from transformers import DonutProcessor, NougatProcessor, TrOCRProcessor
from transformers.testing_utils import ( from transformers.testing_utils import (
is_flaky,
require_levenshtein, require_levenshtein,
require_nltk, require_nltk,
require_sentencepiece, require_sentencepiece,
@@ -323,6 +324,10 @@ class EncoderDecoderMixin:
input_ids_dict = self.prepare_config_and_inputs() input_ids_dict = self.prepare_config_and_inputs()
self.check_encoder_decoder_model_output_attentions(**input_ids_dict) self.check_encoder_decoder_model_output_attentions(**input_ids_dict)
# FIXME @gante: flaky test
@is_flaky(
description="Fails on distributed runs e.g.: https://app.circleci.com/pipelines/github/huggingface/transformers/83611/workflows/666b01c9-1be8-4daa-b85d-189e670fc168/jobs/1078635/tests#failed-test-0"
)
def test_encoder_decoder_model_generate(self): def test_encoder_decoder_model_generate(self):
input_ids_dict = self.prepare_config_and_inputs() input_ids_dict = self.prepare_config_and_inputs()
self.check_encoder_decoder_model_generate(**input_ids_dict) self.check_encoder_decoder_model_generate(**input_ids_dict)