minor bugfix in tests

TuringLang · Mar 2, 2023 · 5fb344e · 5fb344e
1 parent dcf7bc2
commit 5fb344e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/utils.jl b/test/utils.jl
@@ -22,7 +22,7 @@ function StateHistoryCallback(states, selector=deepcopy)
     return StateHistoryCallback{typeof(states), typeof(selector)}(states, selector)
 end
 
-function (cb::StateHistoryCallback)(rng, model, sampler, sample, state, i; kwargs...)
+function (cb::StateHistoryCallback)(rng, model, sampler, transition, state, iteration; kwargs...)
     push!(cb.states, cb.selector(state))
     return nothing
 end