Skip equivalence test for TransfoXL (#16224)

* Skip test for TransfoXL * Single list

Skip equivalence test for TransfoXL (#16224)
* Skip test for TransfoXL * Single list
5a6b3ccd · Lysandre Debut · GitHub · abd503d9 · 5a6b3ccd
Unverified Commit 5a6b3ccd authored Mar 17, 2022 by Lysandre Debut Committed by GitHub Mar 17, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tests/test_modeling_tf_common.py tests/test_modeling_tf_common.py +3 -1

No files found.
--- 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