From c7778bb7c93aa81b4768a10dd3692b5df7509cd6 Mon Sep 17 00:00:00 2001 From: Michal Raczycki Date: Thu, 27 Apr 2023 10:59:37 +0100 Subject: [PATCH] fixed test failing due to infinite precision requirement --- .../tests/distributions/test_discrete_markov_chain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc_experimental/tests/distributions/test_discrete_markov_chain.py b/pymc_experimental/tests/distributions/test_discrete_markov_chain.py index 41d1ef8e..c1458850 100644 --- a/pymc_experimental/tests/distributions/test_discrete_markov_chain.py +++ b/pymc_experimental/tests/distributions/test_discrete_markov_chain.py @@ -93,7 +93,7 @@ def test_logp_with_default_init_dist(self): chain = DiscreteMarkovChain.dist(P=P, init_dist=x0, steps=3) logp = pm.logp(chain, [0, 1, 2]).eval() - assert logp == np.log((1 / 3) * 0.5 * 0.3) + assert logp == pytest.approx(np.log((1 / 3) * 0.5 * 0.3), rel=1e-6) def test_logp_with_user_defined_init_dist(self): P = pt.as_tensor_variable(np.array([[0.1, 0.5, 0.4], [0.3, 0.4, 0.3], [0.9, 0.05, 0.05]]))