Skip to content

GitLab

Unverified Commit 5a6b3ccd authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Skip equivalence test for TransfoXL (#16224) 

* Skip test for TransfoXL

* Single list
parent abd503d9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tests/test_modeling_tf_common.py
View file @ 5a6b3ccd
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment