From 940fde8dafaecb8f17b588c5078291f1c1a420c8 Mon Sep 17 00:00:00 2001
From: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Date: Wed, 5 Jun 2024 19:06:46 +0100
Subject: [PATCH] Skip failing JetMOE generation tests (#31266)

Skip failing tests for now
---
 tests/models/jetmoe/test_modeling_jetmoe.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/models/jetmoe/test_modeling_jetmoe.py b/tests/models/jetmoe/test_modeling_jetmoe.py
index 12e5dd682c6cf5..c61ff01be237c4 100644
--- a/tests/models/jetmoe/test_modeling_jetmoe.py
+++ b/tests/models/jetmoe/test_modeling_jetmoe.py
@@ -472,6 +472,14 @@ def test_flash_attn_2_generate_use_cache(self):
     def test_flash_attn_2_inference_equivalence_right_padding(self):
         self.skipTest("JetMoe flash attention does not support right padding")
 
+    @unittest.skip("TODO: @ArthurZucker - Breaks after #30536 ")
+    def test_beam_sample_generate(self):
+        pass
+
+    @unittest.skip("TODO: @ArthurZucker - Breaks after #30536 ")
+    def test_generate_from_inputs_embeds_decoder_only(self):
+        pass
+
 
 @require_torch
 class JetMoeIntegrationTest(unittest.TestCase):