Fix test

blackjax-devs · Dec 3, 2023 · ae7a1f6 · ae7a1f6
1 parent 448a873
commit ae7a1f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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):