提交 2690f7f0 编写于 作者: P pangyoki

change probs tolerance to 1e-4

上级 32210b08
...@@ -408,7 +408,8 @@ class NormalTest(unittest.TestCase): ...@@ -408,7 +408,8 @@ class NormalTest(unittest.TestCase):
entropy, np_entropy, rtol=tolerance, atol=tolerance) entropy, np_entropy, rtol=tolerance, atol=tolerance)
np.testing.assert_allclose( np.testing.assert_allclose(
log_prob, np_lp, rtol=log_tolerance, atol=log_tolerance) log_prob, np_lp, rtol=log_tolerance, atol=log_tolerance)
np.testing.assert_allclose(probs, np_p, rtol=tolerance, atol=tolerance) np.testing.assert_allclose(
probs, np_p, rtol=log_tolerance, atol=log_tolerance)
np.testing.assert_allclose(kl, np_kl, rtol=tolerance, atol=tolerance) np.testing.assert_allclose(kl, np_kl, rtol=tolerance, atol=tolerance)
def test_normal_distribution_dygraph(self, sample_shape=7, tolerance=1e-6): def test_normal_distribution_dygraph(self, sample_shape=7, tolerance=1e-6):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册