Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/abstractmcmc.jl
Original file line number Diff line number Diff line change
Expand Up @@ -141,6 +141,7 @@ function AbstractMCMC.step(
model::AbstractMCMC.LogDensityModel,
spl::AbstractHMCSampler,
state::HMCState;
n_adapts::Int=0,
kwargs...,
)
# Compute transition.
Expand All @@ -158,7 +159,6 @@ function AbstractMCMC.step(

# Adapt h and spl.
tstat = stat(t)
n_adapts = kwargs[:n_adapts]
h, κ, isadapted = adapt!(h, κ, adaptor, i, n_adapts, t.z.θ, tstat.acceptance_rate)
tstat = merge(tstat, (is_adapt = isadapted,))

Expand Down