Merge pull request #559 from AayushSabharwal/as/rerun-ci

fix: bugs relating to RecursiveArrayTools@3
SciML · Dec 18, 2023 · 7465a4f · 7465a4f
2 parents 083c1e6 + 322375d
commit 7465a4f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -8,6 +8,7 @@ using SciMLBase: ODESolution, sym_to_index, remake,
                  getobserved, build_solution, EnsembleSolution,
                  NonlinearSolution, AbstractTimeseriesSolution
 using SymbolicIndexingInterface: symbolic_type, NotSymbolic
+using RecursiveArrayTools
 
 # This method resolves the ambiguity with the pullback defined in
 # RecursiveArrayToolsZygoteExt
@@ -70,6 +71,9 @@ end
     function EnsembleSolution_adjoint(p̄::AbstractArray{<:AbstractArray, 1})
         (EnsembleSolution(p̄, 0.0, true, stats), nothing, nothing, nothing)
     end
+    function EnsembleSolution_adjoint(p̄::RecursiveArrayTools.AbstractVectorOfArray)
+        (EnsembleSolution(p̄, 0.0, true, stats), nothing, nothing, nothing)
+    end
     function EnsembleSolution_adjoint(p̄::EnsembleSolution)
         (p̄, nothing, nothing, nothing)
     end