Update src/variational/advi.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Jun 10, 2024 · 81f00ef · 81f00ef
1 parent 798bad7
commit 81f00ef
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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)]...)