Implement AbstractMCMC.getstats (#471)

Author: penelopeysm

Project.toml

@@ -1,6 +1,6 @@
 name = "AdvancedHMC"
 uuid = "0bf59076-c3b1-5ca4-86bd-e02cd72cde3d"
-version = "0.8.2"
+version = "0.8.3"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"
@@ -32,7 +32,7 @@ AdvancedHMCOrdinaryDiffEqExt = "OrdinaryDiffEq"
 
 [compat]
 ADTypes = "1"
-AbstractMCMC = "5.6"
+AbstractMCMC = "5.9"
 ArgCheck = "1, 2"
 CUDA = "3, 4, 5"
 ComponentArrays = "0.15"

src/abstractmcmc.jl

@@ -33,6 +33,7 @@ getintegrator(state::HMCState) = state.κ.τ.integrator
 function AbstractMCMC.getparams(state::HMCState)
     return state.transition.z.θ
 end
+AbstractMCMC.getstats(state::AdvancedHMC.HMCState) = state.transition.stat
 
 function AbstractMCMC.setparams!!(
     model::AbstractMCMC.LogDensityModel, state::HMCState, params