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

* try flax fix * same for roberta

[Flax test] Add require pytorch to flix flax test (#8816)
* try flax fix * same for roberta
a2cf3759 · Patrick von Platen · GitHub · e3ef62bc · a2cf3759 · a2cf3759
Unverified Commit a2cf3759 authored Nov 27, 2020 by Patrick von Platen Committed by GitHub Nov 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tests/test_modeling_flax_bert.py tests/test_modeling_flax_bert.py +1 -0

tests/test_modeling_flax_roberta.py tests/test_modeling_flax_roberta.py +1 -0

No files found.
--- a/tests/test_modeling_flax_bert.py
+++ b/tests/test_modeling_flax_bert.py
@@ -48,6 +48,7 @@ class FlaxBertModelTest(unittest.TestCase):


 @require_flax
+@require_torch
 @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
 def test_multiple_sentences(jit):
    tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")

--- a/tests/test_modeling_flax_roberta.py
+++ b/tests/test_modeling_flax_roberta.py
@@ -48,6 +48,7 @@ class FlaxRobertaModelTest(unittest.TestCase):


 @require_flax
+@require_torch
 @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
 def test_multiple_sentences(jit):
    tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")