task format lint

dkweiss31 · Jan 17, 2025 · 001d4cc · 001d4cc
1 parent bcb1f9f
commit 001d4cc
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/floquet/floquet.py b/floquet/floquet.py
@@ -61,15 +61,12 @@ def run_one_floquet(
         fbasis = qt.FloquetBasis(
             self.model.hamiltonian(omega_d_amp),
             T,
-            options={"nsteps": self.options.nsteps}  # type: ignore
+            options={"nsteps": self.options.nsteps},  # type: ignore
         )
         f_modes_0 = fbasis.mode(0.0)
         f_energies = fbasis.e_quasi
         sampling_time = self.options.floquet_sampling_time_fraction * T % T
-        if sampling_time != 0.0:
-            f_modes_t = fbasis.mode(sampling_time)
-        else:
-            f_modes_t = f_modes_0
+        f_modes_t = fbasis.mode(sampling_time) if sampling_time != 0.0 else f_modes_0
         return f_modes_t, f_energies
 
     def identify_floquet_modes(