Support multiple choice in tf common model tests (#4920)

* Support multiple choice in tf common model tests * Add the input_embeds test
2020-06-11 10:31:26 -04:00
parent 699541c4b3
commit 20451195f0
3 changed files with 69 additions and 37 deletions
--- a/tests/test_modeling_tf_bert.py
+++ b/tests/test_modeling_tf_bert.py
@@ -49,6 +49,7 @@ class TFBertModelTest(TFModelTesterMixin, unittest.TestCase):
            TFBertForQuestionAnswering,
            TFBertForSequenceClassification,
            TFBertForTokenClassification,
+            TFBertForMultipleChoice,
        )
        if is_tf_available()
        else ()