@@ -34,9 +34,23 @@ In this notebook we will consider several examples illustrating the usage of the
3434``` python
3535import matplotlib.pyplot as plt
3636import numpy as np
37- from qutip import (QobjEvo, SESolver, UnderDampedEnvironment, about, basis,
38- brmesolve, fidelity, mesolve, qeye, sigmam, sigmax, sigmaz,
39- spost, spre, sprepost)
37+ from qutip import (
38+ QobjEvo,
39+ SESolver,
40+ UnderDampedEnvironment,
41+ about,
42+ basis,
43+ brmesolve,
44+ fidelity,
45+ mesolve,
46+ qeye,
47+ sigmam,
48+ sigmax,
49+ sigmaz,
50+ spost,
51+ spre,
52+ sprepost,
53+ )
4054from qutip.solver.heom import HEOMSolver
4155
4256% matplotlib inline
@@ -531,22 +545,24 @@ adi_me_res = mesolve(H_adi, psi0, tlist, c_ops=c_ops_me, e_ops=e_ops)
531545
532546``` python
533547# HEOM
534- max_depth = 4 # number of hierarchy levels
548+ max_depth = 4 # number of hierarchy levels
535549
536550wsamp = 2 * np.pi
537551w0 = 5 * 2 * np.pi
538552gamma_heom = 1.9 * w0
539553
540554lambd = np.sqrt(
541- 0.5 * gamma / (gamma_heom * wsamp)
555+ 0.5
556+ * gamma
557+ / (gamma_heom * wsamp)
542558 * ((w0** 2 - wsamp** 2 ) ** 2 + (gamma_heom** 2 ) * ((wsamp) ** 2 ))
543559)
544560```
545561
546562``` python
547563# Create Environment
548564bath = UnderDampedEnvironment(lam = lambd, w0 = w0, gamma = gamma_heom, T = 1e-10 )
549- fit_times = np.linspace(0 , 5 , 1000 ) # range for correlation function fit
565+ fit_times = np.linspace(0 , 5 , 1000 ) # range for correlation function fit
550566
551567# Fit correlation function with exponentials
552568exp_bath, fit_info = bath.approx_by_cf_fit(
0 commit comments