Skip to content

Commit 5cf2e18

Browse files
committed
correct mistakes in Flow interface
1 parent 28b8ba0 commit 5cf2e18

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/periodicorbit/FlowDE.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ function dflowMonoSerial(x::AbstractVector, p, dx, tm, pb::ODEProblem, alg; k...
9797
n = length(x)
9898
_prob = remake(pb; u0 = vcat(x, dx), tspan = (zero(tm), tm), p = p)
9999
# the use of concrete_solve makes it compatible with Zygote
100-
sol = SciMLBase.solve(_prob, alg, save_everystep = false; k...)[end]
100+
sol = SciMLBase.solve(_prob, alg; save_everystep = false, k...)[end]
101101
return (t = tm, u = sol[1:n], du = sol[n+1:end])
102102
end
103103

@@ -109,7 +109,7 @@ end
109109

110110
# function used to compute the derivative of the flow, so pb encodes the variational equation
111111
# differential of the flow when a problem is passed for the Monodromy
112-
# default behaviour (the FD case is handled by dispatch)
112+
# default behavior (the FD case is handled by dispatch)
113113
function jvp(fl::FlowDE{T1}, x::AbstractArray, p, dx, tm; kw...) where {T1 <: ODEProblem}
114114
dflowMonoSerial(x, p, dx, tm, fl.probMono, fl.algMono; fl.kwargsDE..., kw...)
115115
end
@@ -159,16 +159,16 @@ function evolve(fl::FlowDE{T1}, ::Val{:SerialTimeSol}, x::AbstractArray, p, tm;
159159
_flow(x, p, tm, fl.prob.prob, fl.alg; fl.kwargsDE..., kw...)
160160
end
161161

162-
function evolve(fl::FlowDE{T1,T2,Tjac,T3}, ::Val{:SerialdFlow}, x::AbstractArray, par, dx, tm; δ = convert(eltype(x), 1e-9), kw...) where {T1 <: ODEProblem,T2,Tjac,T3}
162+
function evolve(fl::FlowDE{T1,T2,Tjac,T3}, ::Val{:SerialdFlow}, x::AbstractArray, par, dx, tm; δ = convert(eltype(x), 1e-9), kw...) where {T1 <: ODEProblem, T2, Tjac, T3}
163163
if T3 === Nothing
164164
return dflow_fdSerial(x, par, dx, tm, fl.prob, fl.alg; δ = δ, fl.kwargsDE..., kw...)
165165
else
166-
return dflowMonoSerial(x, p, dx, tm, fl.probMono, fl.algMono; fl.kwargsDE..., kw...)
166+
return dflowMonoSerial(x, par, dx, tm, fl.probMono, fl.algMono; fl.kwargsDE..., kw...)
167167
end
168168
end
169169

170170
function evolve(fl::FlowDE{T1}, ::Val{:SerialdFlow}, x::AbstractArray, par, dx, tm; kw...) where {T1 <: EnsembleProblem}
171-
dflowMonoSerial(x, p, dx, tm, fl.probMono.prob, fl.algMono; fl.kwargsDE..., kw...)
171+
dflowMonoSerial(x, par, dx, tm, fl.probMono.prob, fl.algMono; fl.kwargsDE..., kw...)
172172
end
173173

174174
function evolve(fl::FlowDE{T1,T2,Tjac,Nothing,T4,T5,T6}, ::Val{:SerialdFlow}, x::AbstractArray, par, dx, tm; δ = convert(eltype(x), 1e-9), kw...) where {T1 <: EnsembleProblem,T2,T4,T5,T6, Tjac}

0 commit comments

Comments
 (0)