Skip to content

Commit 3e64e25

Browse files
chore: rm mentions of SDESolution in ZygoteExt
1 parent 50c2523 commit 3e64e25

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/SciMLBaseZygoteExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,11 +135,11 @@ end
135135
@adjoint function SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u,
136136
args...) where
137137
{uType, tType, isinplace, P, NP, F, G, K, ND}
138-
function SDESolutionAdjoint(ȳ)
138+
function SDEProblemAdjoint(ȳ)
139139
(ȳ, ntuple(_ -> nothing, length(args))...)
140140
end
141141

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

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

0 commit comments

Comments
 (0)