From 5c2d66a2f5da1dc75c2cdc46dd9bc6e7aca4b3b6 Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Tue, 16 Feb 2021 13:59:41 +0100 Subject: [PATCH] Unlock XLA test for convbert (#10207) --- tests/test_modeling_tf_convbert.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/test_modeling_tf_convbert.py b/tests/test_modeling_tf_convbert.py index 69128a55ad..587f3e1dff 100644 --- 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")