Add new dispatch for SDEProblem

Signed-off-by: ErikQQY <[email protected]>
SciML · Aug 17, 2023 · 681b9e1 · 681b9e1
1 parent c6fa298
commit 681b9e1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/problems/sde_problems.jl b/src/problems/sde_problems.jl
@@ -124,13 +124,17 @@ end
 =#
 
 function SDEProblem(f::AbstractSDEFunction, g, u0, tspan, p = NullParameters(); kwargs...)
-    SDEProblem{isinplace(f)}(f, g, u0, tspan, p; kwargs...)
+    SDEProblem{isinplace(f, 4)}(f, g, u0, tspan, p; kwargs...)
 end
 
 function SDEProblem(f, g, u0, tspan, p = NullParameters(); kwargs...)
     SDEProblem(SDEFunction(f, g), g, u0, tspan, p; kwargs...)
 end
 
+function SDEProblem(f::AbstractSDEFunction, u0, tspan, p = NullParameters(); kwargs...)
+    SDEProblem(f, f.g, u0, tspan, p; kwargs...)
+end
+
 """
 $(TYPEDEF)
 """