[testing] LanguageModelGenerationTests require_tf or require_torch (#4616)

2020-05-27 09:10:26 -04:00
parent a9aa7456ac
commit 07797c4da4
11 changed files with 11 additions and 0 deletions
--- a/tests/test_modeling_tf_xlm.py
+++ b/tests/test_modeling_tf_xlm.py
@@ -313,6 +313,7 @@ class TFXLMModelTest(TFModelTesterMixin, unittest.TestCase):
            self.assertIsNotNone(model)


+@require_tf
 class TFXLMModelLanguageGenerationTest(unittest.TestCase):
    @slow
    def test_lm_generate_xlm_mlm_en_2048(self):