fix method ambiguity

TuringLang · Oct 31, 2024 · faf5bb0 · faf5bb0
1 parent 480dedc
commit faf5bb0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/abstractmcmc.jl b/src/abstractmcmc.jl
@@ -34,13 +34,13 @@ function AbstractMCMC.getparams(state::HMCState)
     return state.transition.z.θ
 end
 
-function AbstractMCMC.setparams!!(model, state::HMCState, params)
+function AbstractMCMC.setparams!!(model::AbstractMCMC.LogDensityModel, state::HMCState, params)
     hamiltonian = AdvancedHMC.Hamiltonian(state.metric, model)
     return Setfield.@set state.transition.z = AdvancedHMC.phasepoint(
         hamiltonian,
         params,
         state.transition.z.r;
-        ℓκ = state.transition.z.ℓκ,
+        ℓκ=state.transition.z.ℓκ,
     )
 end