|
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