Merge branch 'feature/bofin-lorentzian-fermionic-baths' into feature/…

…bofin-docs
hodgestar · Dec 2, 2021 · d3b16d4 · d3b16d4
2 parents ce3f774 + 3d4713a
commit d3b16d4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/qutip/nonmarkov/heom.py b/qutip/nonmarkov/heom.py
@@ -28,6 +28,8 @@
     "DrudeLorentzPadeBath",
     "UnderDampedBath",
     "FermionicBath",
+    "LorentzianBath",
+    "LorentzianPadeBath",
     "HEOMSolver",
     "HSolverDL",
 ]
@@ -40,6 +42,8 @@
     DrudeLorentzPadeBath,
     UnderDampedBath,
     FermionicBath,
+    LorentzianBath,
+    LorentzianPadeBath,
 )
 
 from .bofin_solvers import (

diff --git a/qutip/tests/test_heom.py b/qutip/tests/test_heom.py
@@ -10,6 +10,8 @@
     DrudeLorentzPadeBath,
     UnderDampedBath,
     FermionicBath,
+    LorentzianBath,
+    LorentzianPadeBath,
     HEOMSolver,
     HSolverDL,
 )
@@ -25,6 +27,8 @@ def test_api(self):
         assert DrudeLorentzPadeBath
         assert UnderDampedBath
         assert FermionicBath
+        assert LorentzianBath
+        assert LorentzianPadeBath
 
 
 class TestSolverAPI: