bug in pre-allocate convergence history

Hua-Zhou · Hua-Zhou · commit 34ae2ca172de · 2022-04-18T16:13:36.000-07:00
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MultiResponseVarianceComponentModels"
 uuid = "95bd6349-d788-4b5f-978d-e543c2eb4d7d"
 authors = ["Hua Zhou <huazhou@ucla.edu> and contributors"]
-version = "0.1.0"
+version = "0.1.1"
 
 [deps]
 IterativeSolvers = "42fd0dbc-a981-5370-80f2-aaf504508153"
diff --git a/src/fit.jl b/src/fit.jl
@@ -35,9 +35,9 @@ function fit!(
     # record iterate history if requested
     history          = ConvergenceHistory(partial = !log)
     history[:reltol] = reltol
-    IterativeSolvers.reserve!(Int, history, :iter, maxiter)
-    IterativeSolvers.reserve!(T, history, :logl, maxiter)
-    IterativeSolvers.reserve!(Float64, history, :itertime, maxiter)    
+    IterativeSolvers.reserve!(Int    , history, :iter    , maxiter + 1)
+    IterativeSolvers.reserve!(T      , history, :logl    , maxiter + 1)
+    IterativeSolvers.reserve!(Float64, history, :itertime, maxiter + 1)
     # initialization
     tic = time()
     if init == :default
