SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Skip equivalence test for TransfoXL (#16224)

Browse Source 
* Skip test for TransfoXL

* Single list
This commit is contained in:
Lysandre Debut
2022-03-17 14:03:07 +01:00
committed by GitHub
parent abd503d939
commit 5a6b3ccd28
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_tf_common.py
View File

@@ -476,6 +476,7 @@ class TFModelTesterMixin:
"TFFunnelForPreTraining", "TFFunnelForPreTraining",
"TFElectraForPreTraining", "TFElectraForPreTraining",
"TFXLMWithLMHeadModel", "TFXLMWithLMHeadModel",
"TFTransfoXLLMHeadModel",
]: ]:
self.assertEqual(tf_loss is None, pt_loss is None) self.assertEqual(tf_loss is None, pt_loss is None)
@@ -490,7 +491,8 @@ class TFModelTesterMixin:
"TFFunnelForPreTraining", "TFFunnelForPreTraining",
"TFElectraForPreTraining", "TFElectraForPreTraining",
"TFXLMWithLMHeadModel", "TFXLMWithLMHeadModel",
] + ["TFTransfoXLLMHeadModel"]: "TFTransfoXLLMHeadModel",
]:
self.assertEqual(tf_keys, pt_keys) self.assertEqual(tf_keys, pt_keys)
# Since we deliberately make some tests pass above (regarding the `loss`), let's still try to test # Since we deliberately make some tests pass above (regarding the `loss`), let's still try to test