Skip to content

Commit 4a48d01

Browse files
Merge pull request #684 from DhairyaLGandhi/dg/sde2
chore: Remove mentions of SDESolution
2 parents c261a03 + 3e64e25 commit 4a48d01

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

ext/SciMLBaseChainRulesCoreExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -112,15 +112,15 @@ end
112112

113113
function ChainRulesCore.rrule(
114114
::Type{
115-
<:ODESolution{uType, tType, isinplace, P, NP, F, G, K,
115+
<:RODESolution{uType, tType, isinplace, P, NP, F, G, K,
116116
ND
117117
}}, u,
118118
args...) where {uType, tType, isinplace, P, NP, F, G, K, ND}
119-
function SDESolutionAdjoint(ȳ)
119+
function RODESolutionAdjoint(ȳ)
120120
(NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
121121
end
122122

123-
SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint
123+
RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), RODESolutionAdjoint
124124
end
125125

126126
function ChainRulesCore.rrule(::SciMLBase.EnsembleSolution, sim, time, converged)

ext/SciMLBaseZygoteExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,11 +159,11 @@ end
159159
@adjoint function SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u,
160160
args...) where
161161
{uType, tType, isinplace, P, NP, F, G, K, ND}
162-
function SDESolutionAdjoint(ȳ)
162+
function SDEProblemAdjoint(ȳ)
163163
(ȳ, ntuple(_ -> nothing, length(args))...)
164164
end
165165

166-
SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint
166+
SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDEProblemAdjoint
167167
end
168168

169169
@adjoint function NonlinearSolution{T, N, uType, R, P, A, O, uType2}(u,

0 commit comments

Comments
 (0)