Fix test (#9755)

2021-01-22 17:40:16 +01:00
parent 82d46febeb
commit a449ffcbd2
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_tf_common.py
+++ b/tests/test_modeling_tf_common.py
@@ -272,7 +272,7 @@ class TFModelTesterMixin:
                else:
                    output = outputs["attentions"]

-                self.assertEqual(len(output), num_out)
+                self.assertEqual(len(outputs), num_out)
                self.assertEqual(len(output), self.model_tester.num_hidden_layers)
                self.assertListEqual(
                    list(output[0].shape[-3:]),