Merge pull request #21 from JuliaDiffEq/ChrisRackauckas-patch-2

Fix saveat
SciML · Oct 5, 2017 · 3deb10a · 3deb10a
2 parents dbb6543 + 2943da2
commit 3deb10a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/common.jl b/src/common.jl
@@ -57,8 +57,11 @@ function solve{uType,duType,tType,isinplace,LinearSolver}(
     T = tspan[end]
 
     if typeof(saveat) <: Number
-        saveat_vec = convert(Vector{tType},saveat+tspan[1]:saveat:(tspan[end]-saveat))
-        # Exclude the endpoint because of floating point issues
+        if (tspan[1]:saveat:tspan[end])[end] == tspan[end]
+          saveat_vec = convert(Vector{tType},collect(tType,tspan[1]+saveat:saveat:tspan[end]))
+        else
+          saveat_vec = convert(Vector{tType},collect(tType,tspan[1]+saveat:saveat:(tspan[end]-saveat)))
+        end
     else
         saveat_vec =  convert(Vector{tType},collect(saveat))
     end