From 701bad439beb9a12ba36dd6fefeaa3beb1496491 Mon Sep 17 00:00:00 2001 From: ziyu huang Date: Fri, 1 Apr 2022 16:13:29 +0800 Subject: [PATCH] [NFC] polish colossalai/context/process_group_initializer/process_group_initializer.py code stype (#617) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: “Arsmart123 <202476410arsmart@gmail.com> --- .../process_group_initializer.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/colossalai/context/process_group_initializer/process_group_initializer.py b/colossalai/context/process_group_initializer/process_group_initializer.py index 5c16651ad..98150ce8e 100644 --- a/colossalai/context/process_group_initializer/process_group_initializer.py +++ b/colossalai/context/process_group_initializer/process_group_initializer.py @@ -17,14 +17,9 @@ class ProcessGroupInitializer(ABC): pipeline_parallel_size (int): Size of pipeline parallel. tensor_parallel_size (int): Size of tensor parallel. """ - def __init__(self, - rank: int, - world_size: int, - config: Config, - data_parallel_size: int, - pipeline_parallel_size: int, - tensor_parallel_size: int - ): + + def __init__(self, rank: int, world_size: int, config: Config, data_parallel_size: int, pipeline_parallel_size: int, + tensor_parallel_size: int): self.rank = rank self.world_size = world_size self.data_parallel_size = data_parallel_size