fix sparse_attention unittest (#47547)

75b73400 · zhangkaihuo · GitHub · eb100c7b · 75b73400
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_sparse_attention_op.py .../paddle/fluid/tests/unittests/test_sparse_attention_op.py +1 -1

未找到文件。
--- 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):