Skip to content

Commit 1032a6d

Browse files
Reduce the allocations threshold in tests
1 parent d773b0e commit 1032a6d

File tree

3 files changed

+254
-207
lines changed

3 files changed

+254
-207
lines changed

src/time_evolution/callback_helpers/callback_helpers.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ function _generate_save_callback(e_ops, tlist, progress_bar, method)
5757
expvals = e_ops isa Nothing ? nothing : Array{ComplexF64}(undef, length(e_ops), length(tlist))
5858

5959
_save_func = method(e_ops_data, progr, Ref(1), expvals)
60-
return FunctionCallingCallback(_save_func, funcat=tlist)
60+
return FunctionCallingCallback(_save_func, funcat = tlist)
6161
end
6262

6363
function _generate_stochastic_save_callback(e_ops, sc_ops, tlist, store_measurement, progress_bar, method)
@@ -70,7 +70,7 @@ function _generate_stochastic_save_callback(e_ops, sc_ops, tlist, store_measurem
7070
m_expvals = getVal(store_measurement) ? Array{Float64}(undef, length(sc_ops), length(tlist) - 1) : nothing
7171

7272
_save_func = method(store_measurement, e_ops_data, m_ops_data, progr, Ref(1), expvals, m_expvals)
73-
return FunctionCallingCallback(_save_func, funcat=tlist)
73+
return FunctionCallingCallback(_save_func, funcat = tlist)
7474
end
7575

7676
##

src/time_evolution/callback_helpers/mcsolve_callback_helpers.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ function _generate_mcsolve_kwargs(ψ0, T, e_ops, tlist, c_ops, jump_callback, rn
115115
expvals = Array{ComplexF64}(undef, length(e_ops), length(tlist))
116116

117117
_save_func = SaveFuncMCSolve(get_data.(e_ops), Ref(1), expvals)
118-
cb2 = FunctionCallingCallback(_save_func, funcat=tlist)
118+
cb2 = FunctionCallingCallback(_save_func, funcat = tlist)
119119
kwargs2 =
120120
haskey(kwargs, :callback) ? merge(kwargs, (callback = CallbackSet(cb1, cb2, kwargs.callback),)) :
121121
merge(kwargs, (callback = CallbackSet(cb1, cb2),))
@@ -218,7 +218,7 @@ function _mcsolve_initialize_callbacks(cb::CallbackSet, tlist, traj_rng)
218218
e_ops = cb_discrete[idx].affect!.func.e_ops
219219
expvals = similar(cb_discrete[idx].affect!.func.expvals)
220220
_save_func = SaveFuncMCSolve(e_ops, Ref(1), expvals)
221-
cb_save = (FunctionCallingCallback(_save_func, funcat=tlist),)
221+
cb_save = (FunctionCallingCallback(_save_func, funcat = tlist),)
222222
else
223223
cb_save = ()
224224
end
@@ -233,7 +233,7 @@ function _mcsolve_initialize_callbacks(cb::CallbackSet, tlist, traj_rng)
233233
e_ops = cb_discrete[idx].affect!.func.e_ops
234234
expvals = similar(cb_discrete[idx].affect!.func.expvals)
235235
_save_func = SaveFuncMCSolve(e_ops, Ref(1), expvals)
236-
cb_save = (FunctionCallingCallback(_save_func, funcat=tlist),)
236+
cb_save = (FunctionCallingCallback(_save_func, funcat = tlist),)
237237
else
238238
cb_save = ()
239239
end

0 commit comments

Comments
 (0)