From 81f00ef182aede298296818ba3512626791994fa Mon Sep 17 00:00:00 2001
From: Hong Ge <3279477+yebai@users.noreply.github.com>
Date: Mon, 10 Jun 2024 18:17:26 +0100
Subject: [PATCH] Update src/variational/advi.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
---
 src/variational/advi.jl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/variational/advi.jl b/src/variational/advi.jl
index eacf2d01b..ec3e6552e 100644
--- a/src/variational/advi.jl
+++ b/src/variational/advi.jl
@@ -23,8 +23,9 @@ denoting the dimensionality of the latent variables.
 function Bijectors.bijector(
     model::DynamicPPL.Model, ::Val{sym2ranges}=Val(false); varinfo=DynamicPPL.VarInfo(model)
 ) where {sym2ranges}
-    num_params = sum([size(varinfo.metadata[sym].vals, 1) for sym in keys(varinfo.metadata)
-])
+    num_params = sum([
+        size(varinfo.metadata[sym].vals, 1) for sym in keys(varinfo.metadata)
+    ])
 
     dists = vcat([varinfo.metadata[sym].dists for sym in keys(varinfo.metadata)]...)