Skip to content

GitLab

Unverified Commit a2cf3759 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

* try flax fix

* same for roberta
parent e3ef62bc
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/test_modeling_flax_bert.py
View file @ a2cf3759
...@@ -48,6 +48,7 @@ class FlaxBertModelTest(unittest.TestCase): ...@@ -48,6 +48,7 @@ class FlaxBertModelTest(unittest.TestCase):
@require_flax @require_flax
@require_torch
@pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"]) @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
def test_multiple_sentences(jit): def test_multiple_sentences(jit):
tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased") tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")
......
tests/test_modeling_flax_roberta.py
View file @ a2cf3759
...@@ -48,6 +48,7 @@ class FlaxRobertaModelTest(unittest.TestCase): ...@@ -48,6 +48,7 @@ class FlaxRobertaModelTest(unittest.TestCase):
@require_flax @require_flax
@require_torch
@pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"]) @pytest.mark.parametrize("jit", ["disable_jit", "enable_jit"])
def test_multiple_sentences(jit): def test_multiple_sentences(jit):
tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base") tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment