Skip to content

Commit 04ce92a

Browse files
authored
fix: arg order (#349)
Co-authored-by: Kai Xu <[email protected]>
1 parent 1dff529 commit 04ce92a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/src/riemannian_hmc.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ phasepoint(
242242
θ::T,
243243
r::T;
244244
ℓπ = ∂H∂θ(h, θ),
245-
ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, r, θ)),
245+
ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, θ, r)),
246246
) where {T<:AbstractVecOrMat} = PhasePoint(θ, r, ℓπ, ℓκ)
247247

248248
# Negative kinetic energy

0 commit comments

Comments
 (0)