Fix steadystate

Author: albertomercurio

src/time_evolution/time_evolution.jl

@@ -2,8 +2,8 @@ export TimeEvolutionSol, TimeEvolutionMCSol, TimeEvolutionStochasticSol
 
 export liouvillian_floquet, liouvillian_generalized
 
-const DEFAULT_ODE_SOLVER_OPTIONS = (abstol = 1e-8, reltol = 1e-6, save_everystep = false)
-const DEFAULT_SDE_SOLVER_OPTIONS = (abstol = 1e-2, reltol = 1e-2, save_everystep = false)
+const DEFAULT_ODE_SOLVER_OPTIONS = (abstol = 1e-8, reltol = 1e-6, save_everystep = false, save_end = true)
+const DEFAULT_SDE_SOLVER_OPTIONS = (abstol = 1e-2, reltol = 1e-2, save_everystep = false, save_end = true)
 const JUMP_TIMES_WHICH_INIT_SIZE = 200
 
 @doc raw"""
@@ -237,7 +237,13 @@ function _merge_saveat(tlist, e_ops, default_options; kwargs...)
     saveat = is_empty_e_ops ? tlist : [tlist[end]]
     default_values = (default_options..., saveat = saveat)
     kwargs2 = merge(default_values, kwargs)
-    save_end = tlist[end] in kwargs2.saveat # DifferentialEquations.jl has this weird setting
+
+    # DifferentialEquations.jl has this weird save_end setting
+    # So we need to do this to make sure it's consistent
+    haskey(kwargs, :save_end) && return kwargs2
+    isempty(kwargs2.saveat) && return kwargs2
+
+    save_end = tlist[end] in kwargs2.saveat
     return merge(kwargs2, (save_end = save_end,))
 end