From 25c0eb28ccd5088617cebe6b76d2b1cfaab44e7c Mon Sep 17 00:00:00 2001 From: Aurelius84 Date: Fri, 6 Sep 2019 13:58:25 +0800 Subject: [PATCH] fix unittest random failed test=develop (#19654) --- .../paddle/fluid/tests/unittests/test_distributions.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_distributions.py b/python/paddle/fluid/tests/unittests/test_distributions.py index bf001a04ec..3de9c10e6d 100644 --- a/python/paddle/fluid/tests/unittests/test_distributions.py +++ b/python/paddle/fluid/tests/unittests/test_distributions.py @@ -505,8 +505,9 @@ class DistributionTest(unittest.TestCase): feed={'logits': logits_np}, fetch_list=[entropy_np, kl_np]) np.testing.assert_allclose( - output_entropy_np, gt_entropy_np, rtol=tolerance) - np.testing.assert_allclose(output_kl_np, gt_kl_np, rtol=tolerance) + output_entropy_np, gt_entropy_np, rtol=tolerance, atol=tolerance) + np.testing.assert_allclose( + output_kl_np, gt_kl_np, rtol=tolerance, atol=tolerance) def test_multivariateNormalDiag_distribution(self, batch_size=2, @@ -568,8 +569,9 @@ class DistributionTest(unittest.TestCase): }, fetch_list=[entropy_np, kl_np]) np.testing.assert_allclose( - output_entropy_np, gt_entropy_np, rtol=tolerance) - np.testing.assert_allclose(output_kl_np, gt_kl_np, rtol=tolerance) + output_entropy_np, gt_entropy_np, rtol=tolerance, atol=tolerance) + np.testing.assert_allclose( + output_kl_np, gt_kl_np, rtol=tolerance, atol=tolerance) if __name__ == '__main__':