SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

TFRobertaModelIntegrationTest requires tf (#4726)

Browse Source
This commit is contained in:
Sam Shleifer
2020-06-02 12:59:00 -04:00
committed by GitHub
parent d976ef262e
commit 70f7423436
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_tf_roberta.py
View File

@@ -237,6 +237,7 @@ class TFRobertaModelTest(TFModelTesterMixin, unittest.TestCase):
self.assertIsNotNone(model) self.assertIsNotNone(model)
@require_tf
class TFRobertaModelIntegrationTest(unittest.TestCase): class TFRobertaModelIntegrationTest(unittest.TestCase):
@slow @slow
def test_inference_masked_lm(self): def test_inference_masked_lm(self):