From 70f742343602a18c499aaf5d891bec1d379723e0 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 2 Jun 2020 12:59:00 -0400 Subject: [PATCH] TFRobertaModelIntegrationTest requires tf (#4726) --- tests/test_modeling_tf_roberta.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_tf_roberta.py b/tests/test_modeling_tf_roberta.py index 6fbb9a33f6..a7d2603809 100644 --- 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):