bounds bug fix

qutip · May 1, 2024 · 41dd7e2 · 41dd7e2
1 parent 0a96185
commit 41dd7e2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/qutip_qoc/pulse_optim.py b/src/qutip_qoc/pulse_optim.py
@@ -182,9 +182,6 @@ def optimize_pulses(
     qtrl_optimizers = []
     if alg == "CRAB" or alg == "GRAPE":
         if alg == "GRAPE":  # algorithm specific kwargs
-            # only scalar bounds
-            lbound = lbound[0]
-            ubound = ubound[0]
             use_as_amps = True
             alg_params = algorithm_kwargs.get("alg_params", {})
 
@@ -210,6 +207,11 @@ def optimize_pulses(
                 "fix_frequency": fix_frequency,
             }
 
+        if use_as_amps:
+            # same bounds for all controls
+            lbound = lbound[0]
+            ubound = ubound[0]
+
         # one optimizer for each objective
         for i, objective in enumerate(objectives):
             params = {