|
30 | 30 |
|
31 | 31 | function (m::NormalModel)()
|
32 | 32 | # First latent variable.
|
33 |
| - rng = Libtask.get_dynamic_scope() |
| 33 | + rng = Libtask.get_taped_globals(Any) |
34 | 34 | m.a = a = rand(rng, Normal(4, 5))
|
35 | 35 |
|
36 | 36 | # First observation.
|
37 | 37 | AdvancedPS.observe(Normal(a, 2), 3)
|
38 | 38 |
|
39 | 39 | # Second latent variable.
|
40 |
| - rng = Libtask.get_dynamic_scope() |
| 40 | + rng = Libtask.get_taped_globals(Any) |
41 | 41 | m.b = b = rand(rng, Normal(a, 1))
|
42 | 42 |
|
43 | 43 | # Second observation.
|
|
55 | 55 | end
|
56 | 56 |
|
57 | 57 | function (m::FailSMCModel)()
|
58 |
| - rng = Libtask.get_dynamic_scope() |
| 58 | + rng = Libtask.get_taped_globals(Any) |
59 | 59 | m.a = a = rand(rng, Normal(4, 5))
|
60 | 60 |
|
61 |
| - rng = Libtask.get_dynamic_scope() |
| 61 | + rng = Libtask.get_taped_globals(Any) |
62 | 62 | m.b = b = rand(rng, Normal(a, 1))
|
63 | 63 | if a >= 4
|
64 | 64 | AdvancedPS.observe(Normal(b, 2), 1.5)
|
|
82 | 82 |
|
83 | 83 | function (m::TestModel)()
|
84 | 84 | # First hidden variables.
|
85 |
| - rng = Libtask.get_dynamic_scope() |
| 85 | + rng = Libtask.get_taped_globals(Any) |
86 | 86 | m.a = rand(rng, Normal(0, 1))
|
87 | 87 | m.x = x = rand(rng, Bernoulli(1))
|
88 | 88 | m.b = rand(rng, Gamma(2, 3))
|
|
91 | 91 | AdvancedPS.observe(Bernoulli(x / 2), 1)
|
92 | 92 |
|
93 | 93 | # Second hidden variable.
|
94 |
| - rng = Libtask.get_dynamic_scope() |
| 94 | + rng = Libtask.get_taped_globals(Any) |
95 | 95 | m.c = rand(rng, Beta())
|
96 | 96 |
|
97 | 97 | # Second observation.
|
|
167 | 167 | end
|
168 | 168 |
|
169 | 169 | function (m::DummyModel)()
|
170 |
| - rng = Libtask.get_dynamic_scope() |
| 170 | + rng = Libtask.get_taped_globals(Any) |
171 | 171 | m.a = rand(rng, Normal())
|
172 | 172 | AdvancedPS.observe(Normal(), m.a)
|
173 | 173 |
|
174 |
| - rng = Libtask.get_dynamic_scope() |
| 174 | + rng = Libtask.get_taped_globals(Any) |
175 | 175 | m.b = rand(rng, Normal())
|
176 | 176 | return AdvancedPS.observe(Normal(), m.b)
|
177 | 177 | end
|
|
0 commit comments