fix typos

Author: ytdHuang

src/time_evolution/callback_helpers/callback_helpers.jl

@@ -6,8 +6,8 @@ This file contains helper functions for callbacks. The affect! function are defi
 
 abstract type AbstractSaveFunc end
 
-function _merge_tstops(kwargs, prob_is_constant::Bool, tlist)
-    if prob_is_constant
+function _merge_tstops(kwargs, prob_is_const::Bool, tlist)
+    if prob_is_const
         return kwargs
     else
         tstops = haskey(kwargs, :tstops) ? unique!(sort!(vcat(tlist, kwargs.tstops))) : tlist
@@ -35,7 +35,7 @@ function _generate_stochastic_kwargs(
 
     # Ensure that the noise is stored in tlist. # TODO: Fix this directly in DiffEqNoiseProcess.jl
     # See https://github.com/SciML/DiffEqNoiseProcess.jl/issues/214 for example
-    kwargs2 = _merge_tstops(kwargs, true, tlist) # true to force add 'tstops = tlist'
+    kwargs2 = _merge_tstops(kwargs, false, tlist) # set 'prob_is_const = false' to force add 'tstops = tlist'
 
     if SF === SaveFuncSSESolve
         cb_normalize = _ssesolve_generate_normalize_cb()

src/time_evolution/sesolve.jl

@@ -140,7 +140,7 @@ function sesolve(
     H::Union{AbstractQuantumObject{Operator},Tuple},
     ψ0::QuantumObject{Ket},
     tlist::AbstractVector;
-    alg::AbstractODEAlgorithm = Vern7(lazy=false),
+    alg::AbstractODEAlgorithm = Vern7(lazy = false),
     e_ops::Union{Nothing,AbstractVector,Tuple} = nothing,
     params = NullParameters(),
     progress_bar::Union{Val,Bool} = Val(true),
@@ -172,7 +172,7 @@ function sesolve(
     end
 end
 
-function sesolve(prob::TimeEvolutionProblem, alg::AbstractODEAlgorithm = Vern7(lazy=false); kwargs...)
+function sesolve(prob::TimeEvolutionProblem, alg::AbstractODEAlgorithm = Vern7(lazy = false); kwargs...)
     sol = solve(prob.prob, alg; kwargs...)
 
     return _gen_sesolve_solution(sol, prob.times, prob.dimensions)
@@ -227,7 +227,7 @@ function sesolve_map(
     H::Union{AbstractQuantumObject{Operator},Tuple},
     ψ0::AbstractVector{<:QuantumObject{Ket}},
     tlist::AbstractVector;
-    alg::AbstractODEAlgorithm = Vern7(lazy=false),
+    alg::AbstractODEAlgorithm = Vern7(lazy = false),
     ensemblealg::EnsembleAlgorithm = EnsembleThreads(),
     e_ops::Union{Nothing,AbstractVector,Tuple} = nothing,
     params::Union{NullParameters,Tuple} = NullParameters(),
@@ -267,7 +267,7 @@ sesolve_map(H::Union{AbstractQuantumObject{Operator},Tuple}, ψ0::QuantumObject{
 function sesolve_map(
     prob::TimeEvolutionProblem{<:ODEProblem},
     iter::AbstractArray,
-    alg::AbstractODEAlgorithm = Vern7(lazy=false),
+    alg::AbstractODEAlgorithm = Vern7(lazy = false),
     ensemblealg::EnsembleAlgorithm = EnsembleThreads();
     prob_func::Union{Function,Nothing} = nothing,
     output_func::Union{Tuple,Nothing} = nothing,

test/core-test/time_evolution.jl

@@ -134,7 +134,7 @@ end
 
         allocs_tot = @allocations sesolve(H, ψ0, tlist, saveat = [tlist[end]], progress_bar = Val(false)) # Warm-up
         allocs_tot = @allocations sesolve(H, ψ0, tlist, saveat = [tlist[end]], progress_bar = Val(false))
-        @test allocs_tot < 90
+        @test allocs_tot < 95
     end
 
     @testset "Type Inference sesolve" begin