fix a bug

penelopeysm · penelopeysm · commit d4e1edf64b5b · 2025-09-26T18:08:55.000+01:00
diff --git a/src/model.jl b/src/model.jl
@@ -896,7 +896,7 @@ function init!!(
     varinfo::AbstractVarInfo,
     init_strategy::AbstractInitStrategy=InitFromPrior(),
 )
-    new_model = contextualize(model, InitContext(rng, init_strategy))
+    new_model = setleafcontext(model, InitContext(rng, init_strategy))
     return evaluate!!(new_model, varinfo)
 end
 function init!!(

Original file line number	Diff line number	Diff line change
`@@ -896,7 +896,7 @@ function init!!(`
`896`	`896`	`varinfo::AbstractVarInfo,`
`897`	`897`	`init_strategy::AbstractInitStrategy=InitFromPrior(),`
`898`	`898`	`)`
`899`		`- new_model = contextualize(model, InitContext(rng, init_strategy))`
	`899`	`+ new_model = setleafcontext(model, InitContext(rng, init_strategy))`
`900`	`900`	`return evaluate!!(new_model, varinfo)`
`901`	`901`	`end`
`902`	`902`	`function init!!(`