Making TF MPNet model compliant with XLA (#10260)

* Fix XLA * Rework cast * Apply style
2021-02-19 12:56:41 +01:00
parent fb56bf2584
commit 3d72d47f09
2 changed files with 20 additions and 17 deletions
--- a/tests/test_modeling_tf_mpnet.py
+++ b/tests/test_modeling_tf_mpnet.py
@@ -232,10 +232,6 @@ class TFMPNetModelTest(TFModelTesterMixin, unittest.TestCase):
        config_and_inputs = self.model_tester.prepare_config_and_inputs()
        self.model_tester.create_and_check_mpnet_for_token_classification(*config_and_inputs)

-    def test_xla_mode(self):
-        # TODO JP: Make MPNet XLA compliant
-        pass
-
    @slow
    def test_model_from_pretrained(self):
        for model_name in ["microsoft/mpnet-base"]: