Single list

huggingface · Mar 17, 2022 · bb5874a · bb5874a
1 parent 8e6aaa1
commit bb5874a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py
@@ -476,7 +476,8 @@ def check_pt_tf_models(tf_model, pt_model):
                         "TFFunnelForPreTraining",
                         "TFElectraForPreTraining",
                         "TFXLMWithLMHeadModel",
-                    ] + ["TFTransfoXLLMHeadModel"]:
+                        "TFTransfoXLLMHeadModel",
+                    ]:
                         self.assertEqual(tf_loss is None, pt_loss is None)
 
                 tf_keys = tuple([k for k, v in tf_outputs.items() if v is not None])
@@ -490,7 +491,8 @@ def check_pt_tf_models(tf_model, pt_model):
                         "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