Skip to content

Commit 3fe2def

Browse files
Merge pull request #691 from DhairyaLGandhi/dg/cr
fix: correct type params to ODESolution adjoint
2 parents 9b4f6c7 + 8958aea commit 3fe2def

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ext/SciMLBaseChainRulesCoreExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,15 +98,15 @@ end
9898
function ChainRulesCore.rrule(
9999
::Type{
100100
<:ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
101-
T11, T12
101+
T11, T12, T13, T14
102102
}}, u,
103103
args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11,
104-
T12}
104+
T12, T13, T14}
105105
function ODESolutionAdjoint(ȳ)
106106
(NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
107107
end
108108

109-
ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12}(u, args...),
109+
ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14}(u, args...),
110110
ODESolutionAdjoint
111111
end
112112

0 commit comments

Comments
 (0)