From ae7a1f6c8d4bb75ecfaeb08da611b9eb5f6a9bea Mon Sep 17 00:00:00 2001 From: Junpeng Lao Date: Sun, 3 Dec 2023 05:31:17 +0100 Subject: [PATCH] Fix test --- tests/mcmc/test_random_walk_without_chex.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/mcmc/test_random_walk_without_chex.py b/tests/mcmc/test_random_walk_without_chex.py index 938fac2ee..1a7c03676 100644 --- a/tests/mcmc/test_random_walk_without_chex.py +++ b/tests/mcmc/test_random_walk_without_chex.py @@ -106,10 +106,10 @@ class RMHProposalTest(unittest.TestCase): def transition_distribution(self, key, position): return jnp.array([10.0]) - def reject(self, key, previous_proposal, new_proposal): + def reject(self, key, log_p_accept, previous_proposal, new_proposal): return previous_proposal, False, 0.3 - def accept(self, key, previous_proposal, new_proposal): + def accept(self, key, log_p_accept, previous_proposal, new_proposal): return new_proposal, True, 0.5 def compute_ratio(self, new_state, prev_state):