[NFC] polish pipeline process group code style (#2694)

2 years ago · 534f68c83c
1 changed files with 2 additions and 1 deletions
--- a/colossalai/context/process_group_initializer/initializer_pipeline.py
+++ b/colossalai/context/process_group_initializer/initializer_pipeline.py
@ -4,8 +4,9 @@
 from torch import distributed as dist

 from colossalai.registry import DIST_GROUP_INITIALIZER
-from .process_group_initializer import ProcessGroupInitializer
+
 from ..parallel_mode import ParallelMode
+from .process_group_initializer import ProcessGroupInitializer


@DIST_GROUP_INITIALIZER.register_module