Skip to content

Commit

Permalink
file formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
junpenglao committed Apr 5, 2024
1 parent b5c7619 commit d0366ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/examples/howto_reproduce_the_blackjax_image.md
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def smc_inference_loop(loop_key, smc_kernel, init_state, schedule):
i, state = carry
subkey = jax.random.fold_in(loop_key, i)
new_state, info = smc_kernel(subkey, state, lmbda)
return (i+1, new_state), (new_state, info)
return (i + 1, new_state), (new_state, info)

_, (all_samples, _) = jax.lax.scan(body_fn, (0, init_state), schedule)

Expand Down

0 comments on commit d0366ae

Please sign in to comment.