diff --git a/tests/foundationals/latent_diffusion/test_schedulers.py b/tests/foundationals/latent_diffusion/test_schedulers.py
index 845b672..01552f0 100644
--- a/tests/foundationals/latent_diffusion/test_schedulers.py
+++ b/tests/foundationals/latent_diffusion/test_schedulers.py
@@ -14,7 +14,13 @@ def test_scheduler_utils():
         def _generate_timesteps(self) -> Tensor:
             return Tensor()
 
-    scheduler = DummyScheduler(num_inference_steps=10, num_train_timesteps=20, initial_diffusion_rate=0.1, final_diffusion_rate=0.2, device="cpu")
+    scheduler = DummyScheduler(
+        num_inference_steps=10,
+        num_train_timesteps=20,
+        initial_diffusion_rate=0.1,
+        final_diffusion_rate=0.2,
+        device="cpu",
+    )
     scale_factors = (
         1.0
         - linspace(