From 75b73400c4cf8ce77f15a357e1737142ccafe9bc Mon Sep 17 00:00:00 2001 From: zhangkaihuo Date: Wed, 2 Nov 2022 10:28:26 +0800 Subject: [PATCH] fix sparse_attention unittest (#47547) --- python/paddle/fluid/tests/unittests/test_sparse_attention_op.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py b/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py index 92e2d0200c..e10b20a073 100644 --- a/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py +++ b/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py @@ -405,7 +405,7 @@ class TestSparseAttentionAPI(unittest.TestCase): ) np.testing.assert_allclose( - fetches_result, expected_result, rtol=1e-05, atol=1e-05 + fetches_result[0], expected_result, rtol=1e-05, atol=1e-05 ) def test_dygraph(self): -- GitLab