Redefine chain_sym_map

TuringLang · Dec 4, 2024 · 274b46e · 274b46e
1 parent 7672c84
commit 274b46e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/model.jl b/test/model.jl
@@ -62,7 +62,10 @@ const GDEMO_DEFAULT = DynamicPPL.TestUtils.demo_assume_observe_literal()
             logpriors = logprior(model, chain)
             loglikelihoods = loglikelihood(model, chain)
             logjoints = logjoint(model, chain)
-            chain_sym_map = Dict()
+            chain_sym_map = Dict(
+                vn => AbstractPPL.getsym(vn) for vn in DynamicPPL.TestUtils.varnames(model)
+            )
+
             # compare them with true values
             for i in 1:N
                 samples_dict = Dict()