stable working version

all bugs are fixed, but having this in environment.py doesn't make much sense. rework in progress
nwlambert · Dec 11, 2024 · 3d9b5f5 · 3d9b5f5
1 parent 690afd7
commit 3d9b5f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/qutip/core/environment.py b/qutip/core/environment.py
@@ -1756,19 +1756,19 @@ def __init__(
 
         if ck_input is not None:
             exponents.extend(
-                self._make_exponent("Input", 2, Q, ck, 0., tag=tag)
+                self._make_exponent("Input", ck, 0., tag=tag, dim=2)
                 for ck in ck_input
             )
 
         if ck_output_L is not None:
             exponents.extend(
-                self._make_exponent("Output_L", 2, Q, ck, vk, tag=tag)
+                self._make_exponent("Output_L", ck, vk, tag=tag, dim=2)
                 for ck, vk in zip(ck_output_L, vk_output_L)
             )
 
         if ck_output_R is not None:
             exponents.extend(
-                self._make_exponent("Output_R", 2, Q, ck, vk, tag=tag)
+                self._make_exponent("Output_R", ck, vk, tag=tag, dim=2)
                 for ck, vk in zip(ck_output_R, vk_output_R)
             )
         self.exponents = exponents

diff --git a/qutip/solver/heom/bofin_baths.py b/qutip/solver/heom/bofin_baths.py
@@ -241,8 +241,8 @@ class BosonicBath(environment.ExponentialBosonicEnvironment):
     which is not included in the newer "environment" API.
     """
 
-    def _make_exponent(self, type, ck, vk, ck2=None, tag=None):
-        return BathExponent(type, None, self._Q, ck, vk, ck2, tag=tag)
+    def _make_exponent(self, type, ck, vk, ck2=None, tag=None, dim = None):
+        return BathExponent(type, dim, self._Q, ck, vk, ck2, tag=tag)
 
     def _check_coup_op(self, Q):
         if not isinstance(Q, Qobj):