From 72eefb34a9f24f834a8a855ab6e0ed1cc7568af8 Mon Sep 17 00:00:00 2001
From: Lysandre Debut <lysandre@huggingface.co>
Date: Wed, 25 Aug 2021 18:56:25 +0200
Subject: [PATCH] Add require flax to test (#13260)

---
 tests/test_modeling_flax_mt5.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_modeling_flax_mt5.py b/tests/test_modeling_flax_mt5.py
index 8303a1d027..f9ef2d5e18 100644
--- a/tests/test_modeling_flax_mt5.py
+++ b/tests/test_modeling_flax_mt5.py
@@ -15,7 +15,7 @@
 import unittest
 
 from transformers import is_flax_available
-from transformers.testing_utils import require_sentencepiece, require_tokenizers, require_torch, slow
+from transformers.testing_utils import require_flax, require_sentencepiece, require_tokenizers, require_torch, slow
 
 
 if is_flax_available():
@@ -28,6 +28,7 @@ if is_flax_available():
 @require_torch
 @require_sentencepiece
 @require_tokenizers
+@require_flax
 class MT5IntegrationTest(unittest.TestCase):
     @slow
     def test_small_integration_test(self):