Fix test (#9755)

a449ffcb · Julien Plu · GitHub · 82d46feb · a449ffcb
Unverified Commit a449ffcb authored Jan 22, 2021 by Julien Plu Committed by GitHub Jan 22, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- 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:]),