Make black linter conform

qutip · Mar 3, 2024 · cf65a72 · cf65a72
1 parent e92ab51
commit cf65a72
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/qutip_qtrl/pulsegen.py b/src/qutip_qtrl/pulsegen.py
@@ -1292,7 +1292,7 @@ def gen_pulse(self, coeffs=None):
         for i in range(self.num_coeffs):
             if self.fix_freqs:
                 phase = self.freqs[i] * self.time
-            else: # optimise frequencies as part of the parameters
+            else:  # optimise frequencies as part of the parameters
                 phase = self.coeffs[i, 2] * self.time
             pulse += self.coeffs[i, 0] * np.sin(phase) + self.coeffs[
                 i, 1

diff --git a/src/qutip_qtrl/pulseoptim.py b/src/qutip_qtrl/pulseoptim.py
@@ -2042,8 +2042,9 @@ def create_pulse_optimizer(
         optim.pulse_generator = []
         for j in range(n_ctrls):
             crab_pgen = pulsegen.PulseGenCrabFourier(
-                dyn=dyn, num_coeffs=num_coeffs,
-                fix_freqs=alg_params.get("fix_frequency", True)
+                dyn=dyn,
+                num_coeffs=num_coeffs,
+                fix_freqs=alg_params.get("fix_frequency", True),
             )
             if init_coeff_scaling is not None:
                 crab_pgen.scaling = init_coeff_scaling