From a449ffcbd2887b936e6b70a89e533a0bb713743a Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Fri, 22 Jan 2021 17:40:16 +0100 Subject: [PATCH] Fix test (#9755) --- tests/test_modeling_tf_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index 65715f98dc..b41209f7db 100644 --- 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:]),