[NFC] polish colossalai/engine/gradient_handler/utils.py code style (#2708)

2 years ago · 4ac8bfb072
parent 6427c406cf
commit 4ac8bfb072
1 changed files with 30 additions and 29 deletions
--- a/colossalai/engine/gradient_handler/utils.py
+++ b/colossalai/engine/gradient_handler/utils.py
@ -1,7 +1,8 @@
+from typing import Iterable
+
 import torch.distributed as dist
 import torch.nn as nn
 from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
-from typing import Iterable


 def bucket_allreduce(param_list: Iterable[nn.Parameter], group=None):