[Flax test] Add require pytorch to flix flax test (huggingface#8816)

* try flax fix * same for roberta
stas00 · Dec 2, 2020 · fc3d92a · fc3d92a
1 parent 4cb8033
commit fc3d92a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_modeling_flax_bert.py b/tests/test_modeling_flax_bert.py
@@ -48,6 +48,7 @@ def assert_almost_equals(self, a: ndarray, b: ndarray, tol: float):
 
 
 @require_flax
+@require_torch
 @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
 def test_multiple_sentences(jit):
     tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")

diff --git a/tests/test_modeling_flax_roberta.py b/tests/test_modeling_flax_roberta.py
@@ -48,6 +48,7 @@ def assert_almost_equals(self, a: ndarray, b: ndarray, tol: float):
 
 
 @require_flax
+@require_torch
 @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
 def test_multiple_sentences(jit):
     tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")