Update blackjax/mcmc/mala.py

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>
blackjax-devs · Apr 14, 2023 · e2b65b7 · e2b65b7
1 parent 3e8d85c
commit e2b65b7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/blackjax/mcmc/mala.py b/blackjax/mcmc/mala.py
@@ -107,8 +107,7 @@ def one_step(
 
         proposal = init_proposal(state)
         new_proposal, _ = generate_proposal(state, new_state, step_size=step_size)
-        sampled_proposal, *info = sample_proposal(key_rmh, proposal, new_proposal)
-        do_accept, p_accept = info
+        sampled_proposal, do_accept, p_accept = sample_proposal(key_rmh, proposal, new_proposal)
 
         info = MALAInfo(p_accept, do_accept)