From 3e64e258083e7527ee1c8e4e0817e2fee9ac9351 Mon Sep 17 00:00:00 2001 From: DhairyaLGandhi Date: Thu, 2 May 2024 18:31:21 +0530 Subject: [PATCH] chore: rm mentions of SDESolution in ZygoteExt --- ext/SciMLBaseZygoteExt.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl index c7ba1827e..a89bb3a93 100644 --- a/ext/SciMLBaseZygoteExt.jl +++ b/ext/SciMLBaseZygoteExt.jl @@ -135,11 +135,11 @@ end @adjoint function SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...) where {uType, tType, isinplace, P, NP, F, G, K, ND} - function SDESolutionAdjoint(ȳ) + function SDEProblemAdjoint(ȳ) (ȳ, ntuple(_ -> nothing, length(args))...) end - SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint + SDEProblem{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDEProblemAdjoint end @adjoint function NonlinearSolution{T, N, uType, R, P, A, O, uType2}(u,