Update ext/JuliaBUGSMCMCChainsExt.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Oct 10, 2024 · a1eec6e · a1eec6e
1 parent f48109b
commit a1eec6e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/ext/JuliaBUGSMCMCChainsExt.jl b/ext/JuliaBUGSMCMCChainsExt.jl
@@ -69,8 +69,17 @@ function JuliaBUGS.gen_chains(
     generated_quantities = []
     for i in axes(samples)[1]
         evaluation_env = first(evaluate!!(model, LogDensityContext(), samples[i]))
-        push!(param_vals, [AbstractPPL.get(evaluation_env, param_var) for param_var in param_vars])
-        push!(generated_quantities, [AbstractPPL.get(evaluation_env, generated_var) for generated_var in generated_vars])
+        push!(
+            param_vals,
+            [AbstractPPL.get(evaluation_env, param_var) for param_var in param_vars],
+        )
+        push!(
+            generated_quantities,
+            [
+                AbstractPPL.get(evaluation_env, generated_var) for
+                generated_var in generated_vars
+            ],
+        )
     end
 
     param_name_leaves = collect(