Unlock XLA test for convbert (#10207)

5c2d66a2 · Julien Plu · GitHub · 1c8c2d9a · 5c2d66a2
Unverified Commit 5c2d66a2 authored Feb 16, 2021 by Julien Plu Committed by GitHub Feb 16, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 4 deletions

tests/test_modeling_tf_convbert.py tests/test_modeling_tf_convbert.py +0 -4

No files found.
--- a/tests/test_modeling_tf_convbert.py
+++ b/tests/test_modeling_tf_convbert.py
@@ -302,10 +302,6 @@ class TFConvBertModelTest(TFModelTesterMixin, unittest.TestCase):
                    [self.model_tester.num_attention_heads / 2, encoder_seq_length, encoder_key_length],
                )

-    def test_xla_mode(self):
-        # TODO JP: Make ConvBert XLA compliant
-        pass
-
    @slow
    def test_model_from_pretrained(self):
        model = TFConvBertModel.from_pretrained("YituTech/conv-bert-base")