@@ -19,7 +19,7 @@ export Models, to_backend, make_varinfo
1919const SYMBOL_TO_BACKEND = Dict (
2020 :forwarddiff => ADTypes. AutoForwardDiff (),
2121 :reversediff => ADTypes. AutoReverseDiff (; compile= false ),
22- :mooncake => ADTypes. AutoMooncake (; config = nothing ),
22+ :mooncake => ADTypes. AutoMooncake (),
2323)
2424
2525to_backend (x) = error (" Unknown backend: $x " )
@@ -37,8 +37,8 @@ Create a VarInfo for the given `model` using the selected varinfo type.
3737Available varinfo choices:
3838 • `:untyped` → uses `DynamicPPL.untyped_varinfo(model)`
3939 • `:typed` → uses `DynamicPPL.typed_varinfo(model)`
40- • `:simple_namedtuple` → uses `SimpleVarInfo{Float64}(model() )`
41- • `:simple_dict` → builds a `SimpleVarInfo{Float64}` from a Dict (pre-populated with the model’s outputs)
40+ • `:simple_namedtuple` → builds a `SimpleVarInfo{Float64}(::NamedTuple )`
41+ • `:simple_dict` → builds a `SimpleVarInfo{Float64}(::Dict)`
4242
4343The VarInfo is always linked.
4444"""
@@ -50,7 +50,9 @@ function make_varinfo(model::Model, varinfo_choice::Symbol)
5050 elseif varinfo_choice == :typed
5151 DynamicPPL. typed_varinfo (rng, model)
5252 elseif varinfo_choice == :simple_namedtuple
53- SimpleVarInfo {Float64} (model (rng))
53+ vi = DynamicPPL. typed_varinfo (rng, model)
54+ vals = DynamicPPL. values_as (vi, NamedTuple)
55+ SimpleVarInfo {Float64} (vals)
5456 elseif varinfo_choice == :simple_dict
5557 vi = DynamicPPL. typed_varinfo (rng, model)
5658 vals = DynamicPPL. values_as (vi, Dict)
0 commit comments