TFRobertaModelIntegrationTest requires tf (#4726)

2020-06-02 12:59:00 -04:00
parent d976ef262e
commit 70f7423436
1 changed files with 1 additions and 0 deletions
--- a/tests/test_modeling_tf_roberta.py
+++ b/tests/test_modeling_tf_roberta.py
@@ -237,6 +237,7 @@ class TFRobertaModelTest(TFModelTesterMixin, unittest.TestCase):
            self.assertIsNotNone(model)


+@require_tf
 class TFRobertaModelIntegrationTest(unittest.TestCase):
    @slow
    def test_inference_masked_lm(self):