From 04ce92ab167df3c2a309e845911129d3a1d43aa7 Mon Sep 17 00:00:00 2001
From: Kai Xu <5985769+xukai92@users.noreply.github.com>
Date: Wed, 6 Sep 2023 06:48:41 -0400
Subject: [PATCH] fix: arg order (#349)

Co-authored-by: Kai Xu <me@xuk.ai>
---
 research/src/riemannian_hmc.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/research/src/riemannian_hmc.jl b/research/src/riemannian_hmc.jl
index 1bef6355..ab98185e 100644
--- a/research/src/riemannian_hmc.jl
+++ b/research/src/riemannian_hmc.jl
@@ -242,7 +242,7 @@ phasepoint(
     θ::T,
     r::T;
     ℓπ = ∂H∂θ(h, θ),
-    ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, r, θ)),
+    ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, θ, r)),
 ) where {T<:AbstractVecOrMat} = PhasePoint(θ, r, ℓπ, ℓκ)
 
 # Negative kinetic energy