Fix Zygote ODESolution constructor adjoint arg length

SciML · Jun 12, 2024 · 38631ad · 38631ad
1 parent 5575f46
commit 38631ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -220,12 +220,12 @@ end
 @adjoint function ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12
 }(u,
  args...) where {T1, T2, T3, T4, T5, T6, T7, T8,
- T9, T10, T11, T12}
+ T9, T10, T11, T12, T13, T14}
  function ODESolutionAdjoint(ȳ)
  (ȳ, ntuple(_ -> nothing, 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