From 1ce9d0c53102f49ec81e08602c0855ff0816c00f Mon Sep 17 00:00:00 2001 From: RichardoLuo <50363844+RichardoLuo@users.noreply.github.com> Date: Wed, 29 Mar 2023 02:36:29 +0800 Subject: [PATCH] [NFC] polish initializer_data.py code style (#3287) --- .../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 0b8b0d91f..9715ebff7 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