diff --git a/src/sampler.jl b/src/sampler.jl index c0b7cd1f..d66e716e 100644 --- a/src/sampler.jl +++ b/src/sampler.jl @@ -228,7 +228,7 @@ function sample( j = i - drop_warmup * n_adapts if keep_gradients sample = [t.z.θ; t.z.ℓπ] - else + else sample = t.z.θ end θs[j], stats[j] = sample, tstat