better code

Author: penelopeysm

tutorials/variational-inference/index.qmd

@@ -224,13 +224,13 @@ For example, the following callback function estimates the ELBO on `q_avg` every
 
 ```{julia}
 using DynamicPPL: DynamicPPL
+linked_vi = DynamicPPL.link!!(DynamicPPL.VarInfo(m), m);
 
 function callback(; iteration, averaged_params, restructure, kwargs...)
     if mod(iteration, 10) == 1
         q_avg = restructure(averaged_params)
         obj = AdvancedVI.RepGradELBO(128) # 128 samples for ELBO estimation
-        vi = DynamicPPL.link!!(DynamicPPL.VarInfo(m), m);
-        elbo_avg = -estimate_objective(obj, q_avg, LogDensityFunction(m, DynamicPPL.getlogjoint_internal, vi))
+        elbo_avg = -estimate_objective(obj, q_avg, LogDensityFunction(m, DynamicPPL.getlogjoint_internal, linked_vi))
         (elbo_avg = elbo_avg,)
     else
         nothing