Skip equivalence test for TransfoXL (#16224)

* Skip test for TransfoXL * Single list
2022-03-17 14:03:07 +01:00
parent abd503d939
commit 5a6b3ccd28
1 changed files with 3 additions and 1 deletions
--- 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