fixed missing merge in initial step for experimental Gibbs

TuringLang · Jul 10, 2024 · e1f1a0e · e1f1a0e
1 parent 3d3c944
commit e1f1a0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/experimental/gibbs.jl b/src/experimental/gibbs.jl
@@ -308,7 +308,7 @@ function DynamicPPL.initialstep(
     varinfos = map(last, states_and_varinfos)
 
     # Update the base varinfo from the first varinfo and replace it.
-    varinfos_new = DynamicPPL.setindex!!(varinfos, vi_base, 1)
+    varinfos_new = DynamicPPL.setindex!!(varinfos, merge(vi_base, first(varinfos)), 1)
     # Merge the updated initial varinfo with the rest of the varinfos + update the logp.
     vi = DynamicPPL.setlogp!!(
         reduce(merge, varinfos_new),