diff --git a/colossalai/nn/lr_scheduler/__init__.py b/colossalai/nn/lr_scheduler/__init__.py
index fd44686f0..34731ee90 100644
--- a/colossalai/nn/lr_scheduler/__init__.py
+++ b/colossalai/nn/lr_scheduler/__init__.py
@@ -8,6 +8,5 @@ from .torch import LambdaLR, MultiplicativeLR, StepLR, ExponentialLR
 __all__ = [
     'CosineAnnealingLR', 'CosineAnnealingWarmupLR', 'FlatAnnealingLR', 'FlatAnnealingWarmupLR', 'LinearWarmupLR',
     'MultiStepLR', 'MultiStepWarmupLR', 'OneCycleLR', 'PolynomialLR', 'PolynomialWarmupLR', 'LambdaLR',
-    'MultiplicativeLR', 'StepLR',
-    'ExponentialLR'
+    'MultiplicativeLR', 'StepLR', 'ExponentialLR'
 ]