Merge pull request #18 from utkarsh530/saveathandling

Change saveat handling similar to DiffEq
SciML · May 14, 2020 · 43579fd · 43579fd
2 parents f54deb2 + 59d5fe8
commit 43579fd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/MATLABDiffEq.jl b/src/MATLABDiffEq.jl
@@ -29,7 +29,12 @@ function DiffEqBase.__solve(
     f = prob.f
     u0 = prob.u0
 
-    tspan = sort(unique([prob.tspan[1];saveat;prob.tspan[2]]))
+    if typeof(saveat) <: Number
+        tspan = Array(prob.tspan[1]:saveat:prob.tspan[2])
+        tspan = sort(unique([prob.tspan[1];tspan;prob.tspan[2]]))
+    else
+        tspan = sort(unique([prob.tspan[1];saveat;prob.tspan[2]]))
+    end
 
     sizeu = size(prob.u0)