Add missing import

J-Price-3 · J-Price-3 · commit 36bf35d3e3cf · 2025-12-04T12:15:09.000Z
diff --git a/src/riemannian/hamiltonian.jl b/src/riemannian/hamiltonian.jl
@@ -1,7 +1,7 @@
 import AdvancedHMC: refresh, phasepoint, neg_energy, ∂H∂θ, ∂H∂r
 using AdvancedHMC:
     FullMomentumRefreshment, PartialMomentumRefreshment, DualValue, PhasePoint
-using LinearAlgebra: logabsdet, tr, diagm, logdet
+using LinearAlgebra: logabsdet, tr, diagm, logdet, Diagonal
 
 # Specialized phasepoint for Riemannian metrics that need θ for momentum gradient
 function phasepoint(
