Apply suggestions from code review

Co-authored-by: David Widmann <[email protected]>
TuringLang · Oct 4, 2024 · 3b4f6db · 3b4f6db
1 parent 44c55bb
commit 3b4f6db
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/sample.jl b/src/sample.jl
@@ -224,9 +224,6 @@ function mcmcsample(
             # Save the sample.
             samples = save!!(samples, sample, i, model, sampler, N; kwargs...)
 
-            # Increment iteration counter.
-            i += 1
-
             # Update the progress bar.
             if progress && (itotal += 1) >= next_update
                 ProgressLogging.@logprogress itotal / Ntotal
@@ -296,7 +293,7 @@ function mcmcsample(
         # Discard initial samples.
         for j in 1:discard_initial
             # Obtain the next sample and state.
-            sample, state = if j ≤ discard_from_warmup
+            sample, state = if j ≤ num_warmup
                 step_warmup(rng, model, sampler, state; kwargs...)
             else
                 step(rng, model, sampler, state; kwargs...)