Update SciMLBaseZygoteExt.jl

SciML · Oct 28, 2023 · 35f5f06 · 35f5f06
1 parent 30c0342
commit 35f5f06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -56,7 +56,7 @@ end
     VA[sym, j], ODESolution_getindex_pullback
 end
 
-ZygoteRules.@adjoint function DiffEqBase.EnsembleSolution(sim, time, converged, stats)
+ZygoteRules.@adjoint function EnsembleSolution(sim, time, converged, stats)
     out = EnsembleSolution(sim, time, converged)
     function EnsembleSolution_adjoint(p̄::AbstractArray{T, N}) where {T, N}
         arrarr = [[p̄[ntuple(x -> Colon(), Val(N - 2))..., j, i]