From 5a6b3ccd28a320bcde85190b0853ade385bd4158 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 17 Mar 2022 14:03:07 +0100 Subject: [PATCH] Skip equivalence test for TransfoXL (#16224) * Skip test for TransfoXL * Single list --- tests/test_modeling_tf_common.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index 7f46ebe287..2dcda32276 100644 --- a/tests/test_modeling_tf_common.py +++ b/tests/test_modeling_tf_common.py @@ -476,6 +476,7 @@ class TFModelTesterMixin: "TFFunnelForPreTraining", "TFElectraForPreTraining", "TFXLMWithLMHeadModel", + "TFTransfoXLLMHeadModel", ]: self.assertEqual(tf_loss is None, pt_loss is None) @@ -490,7 +491,8 @@ class TFModelTesterMixin: "TFFunnelForPreTraining", "TFElectraForPreTraining", "TFXLMWithLMHeadModel", - ] + ["TFTransfoXLLMHeadModel"]: + "TFTransfoXLLMHeadModel", + ]: self.assertEqual(tf_keys, pt_keys) # Since we deliberately make some tests pass above (regarding the `loss`), let's still try to test