From 8af977f2230af1182a60ceec78a37bf1169765fe Mon Sep 17 00:00:00 2001 From: Arsmart1 <49458769+Arsmart1@users.noreply.github.com> Date: Tue, 28 Mar 2023 14:32:07 +0800 Subject: [PATCH] [NFC] polish colossalai/context/parallel_context.py code style (#3276) --- colossalai/context/parallel_context.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/context/parallel_context.py b/colossalai/context/parallel_context.py index dd12dad6d..0cd533fde 100644 --- a/colossalai/context/parallel_context.py +++ b/colossalai/context/parallel_context.py @@ -10,15 +10,16 @@ from typing import Union import numpy as np import torch import torch.distributed as dist + from colossalai.constants import ALLOWED_MODES, INITIALIZER_MAPPING from colossalai.context.config import Config +from colossalai.context.singleton_meta import SingletonMeta from colossalai.global_variables import tensor_parallel_env as env from colossalai.logging import get_dist_logger from colossalai.registry import DIST_GROUP_INITIALIZER from .parallel_mode import ParallelMode from .random import add_seed, get_seeds, set_mode -from colossalai.context.singleton_meta import SingletonMeta class ParallelContext(metaclass=SingletonMeta):