@@ -19,7 +19,7 @@ export Models, to_backend, make_varinfo
19
19
const SYMBOL_TO_BACKEND = Dict (
20
20
:forwarddiff => ADTypes. AutoForwardDiff (),
21
21
:reversediff => ADTypes. AutoReverseDiff (; compile= false ),
22
- :mooncake => ADTypes. AutoMooncake (; config = nothing ),
22
+ :mooncake => ADTypes. AutoMooncake (),
23
23
)
24
24
25
25
to_backend (x) = error (" Unknown backend: $x " )
@@ -37,8 +37,8 @@ Create a VarInfo for the given `model` using the selected varinfo type.
37
37
Available varinfo choices:
38
38
• `:untyped` → uses `DynamicPPL.untyped_varinfo(model)`
39
39
• `: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)`
42
42
43
43
The VarInfo is always linked.
44
44
"""
@@ -50,7 +50,9 @@ function make_varinfo(model::Model, varinfo_choice::Symbol)
50
50
elseif varinfo_choice == :typed
51
51
DynamicPPL. typed_varinfo (rng, model)
52
52
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)
54
56
elseif varinfo_choice == :simple_dict
55
57
vi = DynamicPPL. typed_varinfo (rng, model)
56
58
vals = DynamicPPL. values_as (vi, Dict)
0 commit comments