From 001d4ccec49b9376cd2a124e17f8bb8a85f7a4c7 Mon Sep 17 00:00:00 2001
From: Danny <dkweiss@u.northwestern.edu>
Date: Thu, 16 Jan 2025 21:24:22 -0500
Subject: [PATCH] task format lint

---
 floquet/floquet.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/floquet/floquet.py b/floquet/floquet.py
index 4b956a3..7e16a0d 100644
--- 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(