From 26da046980839d5114c2c4e524c5efed4b03cdfb Mon Sep 17 00:00:00 2001 From: Gilad Turok <36947659+gil2rok@users.noreply.github.com> Date: Mon, 19 Aug 2024 21:16:56 +0000 Subject: [PATCH] Doc: formatting --- blackjax/vi/fullrank_vi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/blackjax/vi/fullrank_vi.py b/blackjax/vi/fullrank_vi.py index 5e4e23990..321c1fa5c 100644 --- a/blackjax/vi/fullrank_vi.py +++ b/blackjax/vi/fullrank_vi.py @@ -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)