From 77c4c9e474be44abba2002e957b48dcca6becea0 Mon Sep 17 00:00:00 2001
From: Hong Ge <3279477+yebai@users.noreply.github.com>
Date: Wed, 8 Nov 2023 18:46:01 +0000
Subject: [PATCH] Update sampler.jl

---
 src/sampler.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sampler.jl b/src/sampler.jl
index 67925be3e..5c583acd8 100644
--- 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...
     )