From f0aeb1be17b7612f79749ae2a2f7f3ef3ed0c25d Mon Sep 17 00:00:00 2001
From: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Date: Thu, 30 Mar 2023 11:30:19 +0100
Subject: [PATCH] Skip flaky NLLB Moe test for now (#22463)

Skip flaky test for now
---
 tests/models/nllb_moe/test_modeling_nllb_moe.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/models/nllb_moe/test_modeling_nllb_moe.py b/tests/models/nllb_moe/test_modeling_nllb_moe.py
index e2fb1483f7..567aab56ea 100644
--- a/tests/models/nllb_moe/test_modeling_nllb_moe.py
+++ b/tests/models/nllb_moe/test_modeling_nllb_moe.py
@@ -21,6 +21,7 @@ import unittest
 
 from transformers import NllbMoeConfig, is_torch_available, set_seed
 from transformers.testing_utils import (
+    is_flaky,
     require_sentencepiece,
     require_tokenizers,
     require_torch,
@@ -281,6 +282,7 @@ class NllbMoeModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
                 model2, info = model_class.from_pretrained(tmpdirname, output_loading_info=True)
             self.assertEqual(info["missing_keys"], [])
 
+    @is_flaky()
     def test_decoder_model_past_with_large_inputs(self):
         config_and_inputs = self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_decoder_model_past_large_inputs(*config_and_inputs)