From a62d0ea9a5078fb7204179844f96350a47d00867 Mon Sep 17 00:00:00 2001
From: supercooledith <893754954@qq.com>
Date: Mon, 17 Jul 2023 18:30:50 +0800
Subject: [PATCH] [NFC] polish
 colossalai/booster/plugin/low_level_zero_plugin.py code style

---
 colossalai/booster/plugin/low_level_zero_plugin.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/colossalai/booster/plugin/low_level_zero_plugin.py b/colossalai/booster/plugin/low_level_zero_plugin.py
index 94d722080367..3ec0d34092a4 100644
--- a/colossalai/booster/plugin/low_level_zero_plugin.py
+++ b/colossalai/booster/plugin/low_level_zero_plugin.py
@@ -208,10 +208,7 @@ def configure(
 
         if optimizer is not None and \
                 not isinstance(optimizer, OptimizerWrapper):
-            optimizer = LowLevelZeroOptimizer(model.unwrap(),
-                                              optimizer,
-                                              self.zero_optim_config,
-                                              self.optim_kwargs,
+            optimizer = LowLevelZeroOptimizer(model.unwrap(), optimizer, self.zero_optim_config, self.optim_kwargs,
                                               self.verbose)
 
         return model, optimizer, criterion, dataloader, lr_scheduler