adapt to storage for chain starting point quasisample

marius311 · Feb 24, 2020 · 1b366c8 · 1b366c8
1 parent 1bfec09
commit 1b366c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sampling.jl b/src/sampling.jl
@@ -183,7 +183,7 @@ function sample_joint(
         else
             ϕstarts = pmap(θstarts) do θstart
                 Random.seed!()
-                MAP_joint(ds(;θstart...), progress=(progress==:verbose ? :summary : false), Nϕ=Nϕ, quasi_sample=(ϕstart==:quasi_sample); MAP_kwargs...)[2]
+                MAP_joint(adapt(storage,ds(;θstart...)), progress=(progress==:verbose ? :summary : false), Nϕ=adapt(storage,Nϕ), quasi_sample=(ϕstart==:quasi_sample); MAP_kwargs...).ϕ
             end
         end
         chains = pmap(θstarts,ϕstarts) do θstart,ϕstart