[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)

2 years ago · c9e3ee389e
1 changed files with 3 additions and 2 deletions
--- a/colossalai/context/process_group_initializer/initializer_2d.py
+++ b/colossalai/context/process_group_initializer/initializer_2d.py
@ -2,10 +2,11 @@ import math

 import torch.distributed as dist

+from colossalai.global_variables import tensor_parallel_env as env
 from colossalai.registry import DIST_GROUP_INITIALIZER
-from .process_group_initializer import ProcessGroupInitializer
+
 from ..parallel_mode import ParallelMode
-from colossalai.global_variables import tensor_parallel_env as env
+from .process_group_initializer import ProcessGroupInitializer


 def _check_summa_env_var(summa_dim):