From abe4f971e0e316e8558569bf30faca77772367b6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A2=81=E7=88=BD?=
 <100194095+supercooledith@users.noreply.github.com>
Date: Tue, 18 Jul 2023 10:58:43 +0800
Subject: [PATCH] [NFC] polish
 colossalai/booster/plugin/low_level_zero_plugin.py code style (#4256)

Co-authored-by: supercooledith <893754954@qq.com>
---
 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 94d722080..3ec0d3409 100644
--- a/colossalai/booster/plugin/low_level_zero_plugin.py
+++ b/colossalai/booster/plugin/low_level_zero_plugin.py
@@ -208,10 +208,7 @@ class LowLevelZeroPlugin(DPPluginBase):
 
         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