fix: type order

Signed-off-by: Kai Xu <[email protected]>
TuringLang · Jul 23, 2024 · ccfd740 · ccfd740
1 parent 5462aad
commit ccfd740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/riemannian/integrator.jl b/src/riemannian/integrator.jl
@@ -36,7 +36,7 @@ function step(
     n_steps::Int = 1;
     fwd::Bool = n_steps > 0,  # simulate hamiltonian backward when n_steps < 0
     full_trajectory::Val{FullTraj} = Val(false),
-) where {T<:AbstractScalarOrVec{<:AbstractFloat},P<:PhasePoint{TP},FullTraj,TP}
+) where {T<:AbstractScalarOrVec{<:AbstractFloat},TP,P<:PhasePoint{TP},FullTraj}
     n_steps = abs(n_steps)  # to support `n_steps < 0` cases
 
     ϵ = fwd ? step_size(lf) : -step_size(lf)