Update sampler.jl

TuringLang · Nov 8, 2023 · 77c4c9e · 77c4c9e
1 parent 0734d61
commit 77c4c9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sampler.jl b/src/sampler.jl
@@ -87,9 +87,9 @@ function AbstractMCMC.sample(
     N::Integer;
     chain_type=default_chain_type(sampler),
     resume_from=nothing,
+    initial_state = loadstate(resume_from),
     kwargs...,
 )
-    initial_state = loadstate(resume_from)
     return AbstractMCMC.mcmcsample(
         rng, model, sampler, N; chain_type, initial_state, kwargs...
     )