From a2cf37595eab4e297f28d92ed114b8e9f083b190 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 27 Nov 2020 14:40:42 +0100 Subject: [PATCH] [Flax test] Add require pytorch to flix flax test (#8816) * try flax fix * same for roberta --- tests/test_modeling_flax_bert.py | 1 + tests/test_modeling_flax_roberta.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/test_modeling_flax_bert.py b/tests/test_modeling_flax_bert.py index 42d6bfed038..3e9028473d3 100644 --- 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") diff --git a/tests/test_modeling_flax_roberta.py b/tests/test_modeling_flax_roberta.py index f058d54528d..e6a207c291b 100644 --- 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")