@@ -146,8 +146,7 @@ def bench_mesolve(benchmark, model_solve, size):
146
146
elif model_solve == "Qubit Spin Chain" :
147
147
H , psi0 , c_ops , e_ops = qubit_setup (size )
148
148
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 )
151
150
152
151
153
152
def bench_mcsolve (benchmark , model_solve , size ):
@@ -160,8 +159,7 @@ def bench_mcsolve(benchmark, model_solve, size):
160
159
elif model_solve == "Qubit Spin Chain" :
161
160
H , psi0 , c_ops , e_ops = qubit_setup (size )
162
161
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 )
165
163
166
164
167
165
@pytest .mark .nightly
@@ -176,5 +174,4 @@ def bench_steadystate(benchmark, model_steady, size):
176
174
elif model_steady == "Jaynes-Cummings" :
177
175
H , _ , c_ops , _ = jc_setup (size )
178
176
179
- result = benchmark (steadystate , H , c_ops )
180
- return result
177
+ benchmark (steadystate , H , c_ops )
0 commit comments