Skip to content

Commit bea8723

Browse files
committed
Fix deprecation warnings
1 parent f6072b9 commit bea8723

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

test/dynamicppl/varinfo.jl

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,11 @@ using Turing
1414
end
1515

1616
function randr(
17-
vi::DynamicPPL.VarInfo, vn::VarName, dist::Distribution, spl::DynamicPPL.Sampler, count::Bool=false
17+
vi::DynamicPPL.VarInfo,
18+
vn::VarName,
19+
dist::Distribution,
20+
spl::DynamicPPL.Sampler,
21+
count::Bool=false,
1822
)
1923
if !haskey(vi, vn)
2024
r = rand(dist)
@@ -50,9 +54,9 @@ using Turing
5054
alg = HMC(0.1, 5)
5155
spl = DynamicPPL.Sampler(alg, model)
5256
v = copy(meta.vals)
53-
DynamicPPL.link!(vi, spl)
57+
DynamicPPL.link!!(vi, spl, model)
5458
@test all(x -> DynamicPPL.istrans(vi, x), meta.vns)
55-
DynamicPPL.invlink!(vi, spl)
59+
DynamicPPL.invlink!!(vi, spl, model)
5660
@test all(x -> !DynamicPPL.istrans(vi, x), meta.vns)
5761
@test meta.vals == v
5862

@@ -64,10 +68,10 @@ using Turing
6468
@test all(x -> !DynamicPPL.istrans(vi, x), meta.m.vns)
6569
v_s = copy(meta.s.vals)
6670
v_m = copy(meta.m.vals)
67-
DynamicPPL.link!(vi, spl)
71+
DynamicPPL.link!!(vi, spl, model)
6872
@test all(x -> DynamicPPL.istrans(vi, x), meta.s.vns)
6973
@test all(x -> DynamicPPL.istrans(vi, x), meta.m.vns)
70-
DynamicPPL.invlink!(vi, spl)
74+
DynamicPPL.invlink!!(vi, spl, model)
7175
@test all(x -> !DynamicPPL.istrans(vi, x), meta.s.vns)
7276
@test all(x -> !DynamicPPL.istrans(vi, x), meta.m.vns)
7377
@test meta.s.vals == v_s
@@ -347,7 +351,7 @@ using Turing
347351

348352
n = 10
349353
model = state_space(y, length(t))
350-
@test size(sample(model, NUTS(; adtype=AutoReverseDiff(true)), n), 1) == n
354+
@test size(sample(model, NUTS(; adtype=AutoReverseDiff(; compile=true)), n), 1) == n
351355
end
352356

353357
if Threads.nthreads() > 1

0 commit comments

Comments
 (0)