diff --git a/src/model.jl b/src/model.jl index 6d6a23c28..a9947c721 100644 --- a/src/model.jl +++ b/src/model.jl @@ -1162,14 +1162,7 @@ julia> returned(model, Dict{VarName,Float64}(@varname(m) => 2.0)) function returned(model::Model, parameters...) # Note: we can't use `fix(model, parameters)` because # https://github.com/TuringLang/DynamicPPL.jl/issues/1097 - return first( - init!!( - model, - DynamicPPL.OnlyAccsVarInfo(DynamicPPL.AccumulatorTuple()), - # Use `nothing` as the fallback to ensure that any missing parameters cause an - # error - InitFromParams(parameters..., nothing), - UnlinkAll(), - ), - ) + vi = DynamicPPL.OnlyAccsVarInfo(DynamicPPL.AccumulatorTuple()) + init_strategy = InitFromParams(parameters..., nothing) + return first(init!!(model, vi, init_strategy, UnlinkAll())) end