diff --git a/src/sample.jl b/src/sample.jl index cbe5dc87..3bf55083 100644 --- a/src/sample.jl +++ b/src/sample.jl @@ -197,7 +197,7 @@ function mcmcsample( end # Run callback. - callback === nothing || callback(rng, model, sampler, sample, state, 1) + callback === nothing || callback(rng, model, sampler, sample, state, 1; kwargs...) # Save the sample. samples = AbstractMCMC.samples(sample, model, sampler; kwargs...) @@ -217,7 +217,7 @@ function mcmcsample( sample, state = step(rng, model, sampler, state; kwargs...) # Run callback. - callback === nothing || callback(rng, model, sampler, sample, state, i) + callback === nothing || callback(rng, model, sampler, sample, state, i; kwargs...) # Save the sample. samples = save!!(samples, sample, i, model, sampler; kwargs...)