@@ -50,9 +50,9 @@ using Turing
50
50
alg = HMC (0.1 , 5 )
51
51
spl = DynamicPPL. Sampler (alg, model)
52
52
v = copy (meta. vals)
53
- DynamicPPL. link! (vi, spl)
53
+ DynamicPPL. link!! (vi, spl, model )
54
54
@test all (x -> DynamicPPL. istrans (vi, x), meta. vns)
55
- DynamicPPL. invlink! (vi, spl)
55
+ DynamicPPL. invlink!! (vi, spl, model )
56
56
@test all (x -> ! DynamicPPL. istrans (vi, x), meta. vns)
57
57
@test meta. vals == v
58
58
@@ -64,10 +64,10 @@ using Turing
64
64
@test all (x -> ! DynamicPPL. istrans (vi, x), meta. m. vns)
65
65
v_s = copy (meta. s. vals)
66
66
v_m = copy (meta. m. vals)
67
- DynamicPPL. link! (vi, spl)
67
+ DynamicPPL. link!! (vi, spl, model )
68
68
@test all (x -> DynamicPPL. istrans (vi, x), meta. s. vns)
69
69
@test all (x -> DynamicPPL. istrans (vi, x), meta. m. vns)
70
- DynamicPPL. invlink! (vi, spl)
70
+ DynamicPPL. invlink!! (vi, spl, model )
71
71
@test all (x -> ! DynamicPPL. istrans (vi, x), meta. s. vns)
72
72
@test all (x -> ! DynamicPPL. istrans (vi, x), meta. m. vns)
73
73
@test meta. s. vals == v_s
@@ -347,7 +347,7 @@ using Turing
347
347
348
348
n = 10
349
349
model = state_space (y, length (t))
350
- @test size (sample (model, NUTS (; adtype= AutoReverseDiff (true )), n), 1 ) == n
350
+ @test size (sample (model, NUTS (; adtype= AutoReverseDiff (; compile = true )), n), 1 ) == n
351
351
end
352
352
353
353
if Threads. nthreads () > 1
0 commit comments