TFRobertaModelIntegrationTest requires tf (#4726)
This commit is contained in:
parent
d976ef262e
commit
70f7423436
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue