Merge pull request #684 from DhairyaLGandhi/dg/sde2

chore: Remove mentions of SDESolution
SciML · May 2, 2024 · 4a48d01 · 4a48d01
2 parents c261a03 + 3e64e25
commit 4a48d01
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl
@@ -112,15 +112,15 @@ end
 
 function ChainRulesCore.rrule(
         ::Type{
-            <:ODESolution{uType, tType, isinplace, P, NP, F, G, K,
+            <:RODESolution{uType, tType, isinplace, P, NP, F, G, K,
             ND
         }}, u,
         args...) where {uType, tType, isinplace, P, NP, F, G, K, ND}
-    function SDESolutionAdjoint(ȳ)
+    function RODESolutionAdjoint(ȳ)
         (NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
     end
 
-    SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint
+    RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), RODESolutionAdjoint
 end
 
 function ChainRulesCore.rrule(::SciMLBase.EnsembleSolution, sim, time, converged)

diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -159,11 +159,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(ȳ)
+    function SDEProblemAdjoint(ȳ)
         (ȳ, 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,