Update scimlfunctions.jl

SciML · Jul 21, 2024 · bebd61a · bebd61a
1 parent 4363bd7
commit bebd61a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl
@@ -2601,7 +2601,7 @@ end
 
 @add_kwonly function SplitFunction(f1, f2, mass_matrix, cache, analytic, tgrad, jac, jvp,
         vjp, jac_prototype, sparsity, Wfact, Wfact_t, paramjac,
-        observed, colorvec, sys)
+        observed, colorvec, sys, initializeprob, initializeprobmap)
     f1 = ODEFunction(f1)
     f2 = ODEFunction(f2)
 
@@ -2615,8 +2615,10 @@ end
         typeof(cache), typeof(analytic), typeof(tgrad), typeof(jac), typeof(jvp),
         typeof(vjp), typeof(jac_prototype), typeof(sparsity),
         typeof(Wfact), typeof(Wfact_t), typeof(paramjac), typeof(observed), typeof(colorvec),
-        typeof(sys)}(f1, f2, mass_matrix, cache, analytic, tgrad, jac, jvp, vjp,
-        jac_prototype, sparsity, Wfact, Wfact_t, paramjac, observed, colorvec, sys)
+        typeof(sys), typeof(initializeprob), typeof(initializeprobmap}(f1, f2, mass_matrix, 
+        cache, analytic, tgrad, jac, jvp, vjp,
+        jac_prototype, sparsity, Wfact, Wfact_t, paramjac, observed, colorvec, sys,
+        initializeprob, initializeprobmap)
 end
 function SplitFunction{iip, specialize}(f1, f2;
         mass_matrix = __has_mass_matrix(f1) ?