Skip to content

Commit 39a7726

Browse files
committed
Fix essential/container tests
1 parent 6aaad1f commit 39a7726

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

test/essential/container.jl

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,19 +9,26 @@ using Turing
99
@testset "container.jl" begin
1010
@model function test()
1111
a ~ Normal(0, 1)
12+
println("a")
1213
x ~ Bernoulli(1)
14+
println("x")
1315
b ~ Gamma(2, 3)
16+
println("b")
1417
1 ~ Bernoulli(x / 2)
18+
println("1")
1519
c ~ Beta()
20+
println("c")
1621
0 ~ Bernoulli(x / 2)
22+
println("0")
1723
return x
1824
end
1925

2026
@testset "constructor" begin
2127
vi = DynamicPPL.VarInfo()
22-
sampler = Sampler(PG(10))
28+
vi = DynamicPPL.setacc!!(vi, Turing.Inference.ProduceLogLikelihoodAccumulator())
29+
spl = Sampler(PG(10))
2330
model = test()
24-
trace = AdvancedPS.Trace(model, sampler, vi, AdvancedPS.TracedRNG())
31+
trace = AdvancedPS.Trace(model, spl, vi, AdvancedPS.TracedRNG())
2532

2633
# Make sure the backreference from taped_globals to the trace is in place.
2734
@test trace.model.ctask.taped_globals.other === trace
@@ -46,6 +53,7 @@ using Turing
4653
return a, b
4754
end
4855
vi = DynamicPPL.VarInfo()
56+
vi = DynamicPPL.setacc!!(vi, Turing.Inference.ProduceLogLikelihoodAccumulator())
4957
sampler = Sampler(PG(10))
5058
model = normal()
5159

0 commit comments

Comments
 (0)