Merge pull request #85 from TuringLang/kx/fix-resize

Fix resize
TuringLang · Jul 26, 2019 · 48287bc · 48287bc
2 parents e09185d + 4e123a1
commit 48287bc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/adaptation/precond.jl b/src/adaptation/precond.jl
@@ -227,7 +227,7 @@ function Base.resize!(
     if length(θ) != length(dpc.var)
         @assert dpc.ve.n == 0 "Cannot resize a var estimator when it contains samples."
         dpc.ve = WelfordVar(T, length(θ))
-        dpc.var = zeros(T, length(θ))
+        dpc.var = ones(T, length(θ))
     end
 end
 function Base.resize!(

diff --git a/test/adaptation/precond.jl b/test/adaptation/precond.jl
@@ -49,7 +49,7 @@ end
     AdvancedHMC.adapt!(pc1, θ, 1.)
     AdvancedHMC.adapt!(pc2, θ, 1.)
     AdvancedHMC.adapt!(pc3, θ, 1.)
-    @test AdvancedHMC.Adaptation.getM⁻¹(pc2) == zeros(length(θ))
+    @test AdvancedHMC.Adaptation.getM⁻¹(pc2) == ones(length(θ))
     @test AdvancedHMC.Adaptation.getM⁻¹(pc3) == LinearAlgebra.diagm(0 => ones(length(θ)))
 end
 
@@ -75,6 +75,6 @@ end
     AdvancedHMC.adapt!(adaptor1, θ, 1.)
     AdvancedHMC.adapt!(adaptor2, θ, 1.)
     AdvancedHMC.adapt!(adaptor3, θ, 1.)
-    @test AdvancedHMC.Adaptation.getM⁻¹(adaptor2) == zeros(length(θ))
+    @test AdvancedHMC.Adaptation.getM⁻¹(adaptor2) == ones(length(θ))
     @test AdvancedHMC.Adaptation.getM⁻¹(adaptor3) == LinearAlgebra.diagm(0 => ones(length(θ)))
 end