Implement getstepsize() for NoAdaptation samplers

TuringLang · Nov 29, 2024 · 406c2ad · 406c2ad
1 parent f3b2476
commit 406c2ad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/mcmc/hmc.jl b/src/mcmc/hmc.jl
@@ -463,6 +463,12 @@ end
 
 getstepsize(sampler::Sampler{<:Hamiltonian}, state) = sampler.alg.ϵ
 getstepsize(sampler::Sampler{<:AdaptiveHamiltonian}, state) = AHMC.getϵ(state.adaptor)
+function getstepsize(
+    sampler::Sampler{<:AdaptiveHamiltonian},
+    state::HMCState{TV,TKernel,THam,PhType,AHMC.Adaptation.NoAdaptation},
+) where {TV,TKernel,THam,PhType}
+    return state.kernel.τ.integrator.ϵ
+end
 
 gen_metric(dim::Int, spl::Sampler{<:Hamiltonian}, state) = AHMC.UnitEuclideanMetric(dim)
 function gen_metric(dim::Int, spl::Sampler{<:AdaptiveHamiltonian}, state)