Merge pull request #691 from DhairyaLGandhi/dg/cr

fix: correct type params to ODESolution adjoint
SciML · May 13, 2024 · 3fe2def · 3fe2def
2 parents 9b4f6c7 + 8958aea
commit 3fe2def
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl
@@ -98,15 +98,15 @@ end
 function ChainRulesCore.rrule(
         ::Type{
             <:ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
-            T11, T12
+            T11, T12, T13, T14
         }}, u,
         args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11,
-        T12}
+        T12, T13, T14}
     function ODESolutionAdjoint(ȳ)
         (NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
     end
 
-    ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12}(u, args...),
+    ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14}(u, args...),
     ODESolutionAdjoint
 end