Fix more tests

Author: penelopeysm

src/sampler.jl

@@ -59,8 +59,8 @@ function AbstractMCMC.step(
 )
     vi = VarInfo()
     strategy = sampler isa SampleFromPrior ? PriorInit() : UniformInit()
-    DynamicPPL.init!!(rng, model, vi, strategy)
-    return vi, nothing
+    _, new_vi = DynamicPPL.init!!(rng, model, vi, strategy)
+    return new_vi, nothing
 end
 
 """

test/sampler.jl

@@ -25,8 +25,8 @@
         @test length(chains) == N
 
         # `m` is Gaussian, i.e. no transformation is used, so it
-        # should have a mean equal to its prior, i.e. 2.
-        @test mean(vi[@varname(m)] for vi in chains) ≈ 2 atol = 0.1
+        # will be drawn from U[-2, 2] and its mean should be 0.
+        @test mean(vi[@varname(m)] for vi in chains) ≈ 0.0 atol = 0.1
 
         # Expected value of ``exp(X)`` where ``X ~ U[-2, 2]`` is ≈ 1.8.
         @test mean(vi[@varname(s)] for vi in chains) ≈ 1.8 atol = 0.1