Skip to content

Commit

Permalink
Mark test_encoder_decoder_model_generate for `vision_encoder_deocde…
Browse files Browse the repository at this point in the history
…r` as flaky (huggingface#28842)

Mark test as flaky
  • Loading branch information
amyeroberts authored Feb 2, 2024
1 parent 80d5007 commit 3d2900e
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@

from transformers import DonutProcessor, NougatProcessor, TrOCRProcessor
from transformers.testing_utils import (
is_flaky,
require_levenshtein,
require_nltk,
require_sentencepiece,
Expand Down Expand Up @@ -323,6 +324,10 @@ def test_encoder_decoder_model_output_attentions(self):
input_ids_dict = self.prepare_config_and_inputs()
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):
input_ids_dict = self.prepare_config_and_inputs()
self.check_encoder_decoder_model_generate(**input_ids_dict)
Expand Down

0 comments on commit 3d2900e

Please sign in to comment.