dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

* try flax fix

* same for roberta
This commit is contained in:
Patrick von Platen 2020-11-27 14:40:42 +01:00 committed by GitHub
parent e3ef62bce1
commit a2cf37595e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_flax_bert.py
View File

@ -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")

1
tests/test_modeling_flax_roberta.py
View File

@ -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")