|
9 | 9 | )
|
10 | 10 | if !haskey(vi, vn)
|
11 | 11 | r = rand(dist)
|
12 |
| - push!(vi, vn, r, dist, spl) |
| 12 | + push!!(vi, vn, r, dist, spl) |
13 | 13 | r
|
14 | 14 | elseif is_flagged(vi, vn, "del")
|
15 | 15 | unset_flag!(vi, vn, "del")
|
|
125 | 125 | @test vi.metadata.orders == [1, 1, 2, 2, 3, 3]
|
126 | 126 | @test get_num_produce(vi) == 3
|
127 | 127 |
|
128 |
| - vi = empty!(TypedVarInfo(vi)) |
| 128 | + vi = empty!!(TypedVarInfo(vi)) |
129 | 129 | # First iteration, variables are added to vi
|
130 | 130 | # variables samples in order: z1,a1,z2,a2,z3
|
131 | 131 | increment_num_produce!(vi)
|
|
211 | 211 | dists = [Normal(0, 1), MvNormal(zeros(2), I), Wishart(7, [1 0.5; 0.5 1])]
|
212 | 212 | function test_varinfo!(vi)
|
213 | 213 | @test getlogp(vi) === 0.0
|
214 |
| - setlogp!(vi, 1) |
| 214 | + vi = setlogp!!(vi, 1) |
215 | 215 | @test getlogp(vi) === 1.0
|
216 |
| - acclogp!(vi, 1) |
| 216 | + vi = acclogp!!(vi, 1) |
217 | 217 | @test getlogp(vi) === 2.0
|
218 |
| - resetlogp!(vi) |
| 218 | + vi = resetlogp!!(vi) |
219 | 219 | @test getlogp(vi) === 0.0
|
220 | 220 |
|
221 | 221 | spl2 = Sampler(PG(5, :w, :u), empty_model())
|
|
263 | 263 | end
|
264 | 264 | vi = VarInfo()
|
265 | 265 | test_varinfo!(vi)
|
266 |
| - test_varinfo!(empty!(TypedVarInfo(vi))) |
| 266 | + test_varinfo!(empty!!(TypedVarInfo(vi))) |
267 | 267 |
|
268 | 268 | @model igtest() = begin
|
269 | 269 | x ~ InverseGamma(2, 3)
|
|
299 | 299 |
|
300 | 300 | g = Sampler(Gibbs(PG(10, :x, :y, :z), HMC(0.4, 8, :w, :u)), g_demo_f)
|
301 | 301 | pg, hmc = g.state.samplers
|
302 |
| - vi = empty!(TypedVarInfo(vi)) |
| 302 | + vi = empty!!(TypedVarInfo(vi)) |
303 | 303 | @inferred g_demo_f(vi, SampleFromPrior())
|
304 | 304 | pg.state.vi = vi
|
305 | 305 | step!(Random.GLOBAL_RNG, g_demo_f, pg, 1)
|
|
0 commit comments