Add more models to common tests (#4910)

2020-06-10 13:19:53 -04:00
parent 3b3619a327
commit 4e10acb3e5
9 changed files with 59 additions and 9 deletions
--- a/tests/test_modeling_longformer.py
+++ b/tests/test_modeling_longformer.py
@@ -296,7 +296,19 @@ class LongformerModelTest(ModelTesterMixin, unittest.TestCase):
    test_headmasking = False  # head masking is not supported
    test_torchscript = False

-    all_model_classes = (LongformerModel, LongformerForMaskedLM,) if is_torch_available() else ()
+    all_model_classes = (
+        (
+            LongformerModel,
+            LongformerForMaskedLM,
+            # TODO: make tests pass for those models
+            # LongformerForSequenceClassification,
+            # LongformerForQuestionAnswering,
+            # LongformerForTokenClassification,
+            # LongformerForMultipleChoice,
+        )
+        if is_torch_available()
+        else ()
+    )

    def setUp(self):
        self.model_tester = LongformerModelTester(self)