minor update

Author: ytdHuang

src/time_evolution/sesolve.jl

@@ -210,7 +210,7 @@ function sesolve_map(
     )
 
     # generate and solve ensemble problem
-    _output_func = _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj) # setup global progress bar
+    _output_func = _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj, _dummy_output_func) # setup global progress bar
     ens_prob = TimeEvolutionProblem(
         EnsembleProblem(
             prob.prob,

src/time_evolution/smesolve.jl

@@ -251,7 +251,7 @@ function smesolveEnsembleProblem(
         ) : prob_func
     _output_func =
         output_func isa Nothing ?
-        _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj, _stochastic_output_func) : output_func
+        _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj, _dummy_output_func) : output_func
 
     prob_sme = smesolveProblem(
         H,

src/time_evolution/ssesolve.jl

@@ -245,7 +245,7 @@ function ssesolveEnsembleProblem(
         ) : prob_func
     _output_func =
         output_func isa Nothing ?
-        _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj, _stochastic_output_func) : output_func
+        _ensemble_dispatch_output_func(ensemblealg, progress_bar, ntraj, _dummy_output_func) : output_func
 
     prob_sme = ssesolveProblem(
         H,

src/time_evolution/time_evolution.jl

@@ -357,15 +357,16 @@ Helpers for handling output of ensemble problems.
 This is very useful especially for dispatching which method to use to update the progress bar.
 =#
 
+# a dummy output function which does nothing
+_dummy_output_func(sol, i) = (sol, false)
+
 # Output function with progress bar update
-_ensemble_output_func_progress(sol, i, progr, ::Nothing) = next!(progr)
 function _ensemble_output_func_progress(sol, i, progr, output_func)
     next!(progr)
     return output_func(sol, i)
 end
 
 # Output function with distributed channel update for progress bar
-_ensemble_output_func_distributed(sol, i, channel, ::Nothing) = put!(channel, true)
 function _ensemble_output_func_distributed(sol, i, channel, output_func)
     put!(channel, true)
     return output_func(sol, i)
@@ -375,7 +376,7 @@ function _ensemble_dispatch_output_func(
     ::ET,
     progress_bar,
     ntraj,
-    output_func = nothing,
+    output_func,
 ) where {ET<:Union{EnsembleSerial,EnsembleThreads}}
     if getVal(progress_bar)
         progr = ProgressBar(ntraj, enable = getVal(progress_bar))
@@ -389,7 +390,7 @@ function _ensemble_dispatch_output_func(
     ::ET,
     progress_bar,
     ntraj,
-    output_func = nothing,
+    output_func,
 ) where {ET<:Union{EnsembleSplitThreads,EnsembleDistributed}}
     if getVal(progress_bar)
         progr = ProgressBar(ntraj, enable = getVal(progress_bar))
@@ -454,9 +455,6 @@ function _stochastic_prob_func(prob, i, repeat, rng, seeds, tlist; kwargs...)
     return remake(prob.prob, noise = noise, seed = seed)
 end
 
-# Standard output function
-_stochastic_output_func(sol, i) = (sol, false)
-
 #= 
     Define diagonal or non-diagonal noise depending on the type of `sc_ops`.
     If `sc_ops` is a `AbstractQuantumObject`, we avoid using the non-diagonal noise.