From 9e0078d7cf24b01d57bd92d9e354e5bad767a40e Mon Sep 17 00:00:00 2001 From: RichardoLuo <1404955596@qq.com> Date: Tue, 28 Mar 2023 23:17:35 +0800 Subject: [PATCH] [NFC] polish initializer_data.py code style --- .../context/process_group_initializer/initializer_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/context/process_group_initializer/initializer_data.py b/colossalai/context/process_group_initializer/initializer_data.py index 0b8b0d91fcb9..9715ebff7f00 100644 --- a/colossalai/context/process_group_initializer/initializer_data.py +++ b/colossalai/context/process_group_initializer/initializer_data.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