Skip to content

Commit 41125c4

Browse files
committed
Fix calls to get_dynamic_scope
1 parent d7dff1c commit 41125c4

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

test/smc.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,14 @@
3030

3131
function (m::NormalModel)()
3232
# First latent variable.
33-
rng = Libtask.get_dynamic_scope()
33+
rng = Libtask.get_taped_globals(Any)
3434
m.a = a = rand(rng, Normal(4, 5))
3535

3636
# First observation.
3737
AdvancedPS.observe(Normal(a, 2), 3)
3838

3939
# Second latent variable.
40-
rng = Libtask.get_dynamic_scope()
40+
rng = Libtask.get_taped_globals(Any)
4141
m.b = b = rand(rng, Normal(a, 1))
4242

4343
# Second observation.
@@ -55,10 +55,10 @@
5555
end
5656

5757
function (m::FailSMCModel)()
58-
rng = Libtask.get_dynamic_scope()
58+
rng = Libtask.get_taped_globals(Any)
5959
m.a = a = rand(rng, Normal(4, 5))
6060

61-
rng = Libtask.get_dynamic_scope()
61+
rng = Libtask.get_taped_globals(Any)
6262
m.b = b = rand(rng, Normal(a, 1))
6363
if a >= 4
6464
AdvancedPS.observe(Normal(b, 2), 1.5)
@@ -82,7 +82,7 @@
8282

8383
function (m::TestModel)()
8484
# First hidden variables.
85-
rng = Libtask.get_dynamic_scope()
85+
rng = Libtask.get_taped_globals(Any)
8686
m.a = rand(rng, Normal(0, 1))
8787
m.x = x = rand(rng, Bernoulli(1))
8888
m.b = rand(rng, Gamma(2, 3))
@@ -91,7 +91,7 @@
9191
AdvancedPS.observe(Bernoulli(x / 2), 1)
9292

9393
# Second hidden variable.
94-
rng = Libtask.get_dynamic_scope()
94+
rng = Libtask.get_taped_globals(Any)
9595
m.c = rand(rng, Beta())
9696

9797
# Second observation.
@@ -167,11 +167,11 @@
167167
end
168168

169169
function (m::DummyModel)()
170-
rng = Libtask.get_dynamic_scope()
170+
rng = Libtask.get_taped_globals(Any)
171171
m.a = rand(rng, Normal())
172172
AdvancedPS.observe(Normal(), m.a)
173173

174-
rng = Libtask.get_dynamic_scope()
174+
rng = Libtask.get_taped_globals(Any)
175175
m.b = rand(rng, Normal())
176176
return AdvancedPS.observe(Normal(), m.b)
177177
end

0 commit comments

Comments
 (0)