diff --git a/colossalai/nn/lr_scheduler/torch.py b/colossalai/nn/lr_scheduler/torch.py
index 6b57d18f2..05d2a49c1 100644
--- a/colossalai/nn/lr_scheduler/torch.py
+++ b/colossalai/nn/lr_scheduler/torch.py
@@ -58,8 +58,7 @@ class StepLR(_StepLR):
     """
 
     def __init__(self, optimizer, total_steps, step_size: int = 1, gamma: float = 0.1, last_epoch: int = -1) -> None:
-        super().__init__(optimizer, step_size,
-                         gamma=gamma, last_epoch=last_epoch)
+        super().__init__(optimizer, step_size, gamma=gamma, last_epoch=last_epoch)
 
 
 @LR_SCHEDULERS.register_module
@@ -74,6 +73,5 @@ class ExponentialLR(_ExponentialLR):
         last_epoch (int, optional): The index of last epoch, defaults to -1.
     """
 
-    def __init__(self, optimizer, total_steps, gamma: float = 1.0,
-                 last_epoch: int = -1) -> None:
+    def __init__(self, optimizer, total_steps, gamma: float = 1.0, last_epoch: int = -1) -> None:
         super().__init__(optimizer, gamma, last_epoch=last_epoch)