Deal with multiple choice in common tests (#4886)

* Deal with multiple choice in common tests
2020-06-10 08:10:20 -04:00
parent e8db8b845a
commit 0a375f5abd
2 changed files with 29 additions and 17 deletions
--- a/tests/test_modeling_bert.py
+++ b/tests/test_modeling_bert.py
@@ -407,6 +407,7 @@ class BertModelTest(ModelTesterMixin, unittest.TestCase):
        (
            BertModel,
            BertForMaskedLM,
+            BertForMultipleChoice,
            BertForNextSentencePrediction,
            BertForPreTraining,
            BertForQuestionAnswering,