Fix test (huggingface#9755)

sorami · Jan 22, 2021 · a449ffc · a449ffc
1 parent 82d46fe
commit a449ffc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py
@@ -272,7 +272,7 @@ def test_saved_model_with_attentions_output(self):
                 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:]),