Skip to content

Commit

Permalink
Doc: formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
gil2rok committed Aug 19, 2024
1 parent ced702f commit 26da046
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion blackjax/vi/fullrank_vi.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ def _sample(rng_key, mu, chol_params, num_samples):
Samples drawn from the full-rank Gaussian approximation.
"""

mu_flatten, unravel_fn = jax.flatten_util.ravel_pytree(mu)
dim = mu_flatten.size
chol_factor = _unflatten_cholesky(chol_params, dim)
Expand Down

0 comments on commit 26da046

Please sign in to comment.