@@ -146,8 +146,7 @@ def bench_mesolve(benchmark, model_solve, size):
146146 elif model_solve == "Qubit Spin Chain" :
147147 H , psi0 , c_ops , e_ops = qubit_setup (size )
148148
149- result = benchmark (mesolve , H , psi0 , tlist , c_ops , e_ops )
150- return result
149+ benchmark (mesolve , H , psi0 , tlist , c_ops , e_ops = e_ops )
151150
152151
153152def bench_mcsolve (benchmark , model_solve , size ):
@@ -160,8 +159,7 @@ def bench_mcsolve(benchmark, model_solve, size):
160159 elif model_solve == "Qubit Spin Chain" :
161160 H , psi0 , c_ops , e_ops = qubit_setup (size )
162161
163- result = benchmark (mcsolve , H , psi0 , tlist , c_ops , e_ops , ntraj = 1 )
164- return result
162+ benchmark (mcsolve , H , psi0 , tlist , c_ops , e_ops = e_ops , ntraj = 1 )
165163
166164
167165@pytest .mark .nightly
@@ -176,5 +174,4 @@ def bench_steadystate(benchmark, model_steady, size):
176174 elif model_steady == "Jaynes-Cummings" :
177175 H , _ , c_ops , _ = jc_setup (size )
178176
179- result = benchmark (steadystate , H , c_ops )
180- return result
177+ benchmark (steadystate , H , c_ops )
0 commit comments