From a50d39a143c8ae83e1ae4960d2026bb53199d3ad Mon Sep 17 00:00:00 2001 From: dayellow <49357110+dayellow@users.noreply.github.com> Date: Tue, 18 Jul 2023 13:51:37 +0800 Subject: [PATCH] [NFC] fix: format (#4270) * [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style * [NFC] polish colossalai/communication/utils.py code style --------- Co-authored-by: Minghao Huang --- colossalai/communication/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/communication/utils.py b/colossalai/communication/utils.py index ef9eceea8..1516df356 100644 --- a/colossalai/communication/utils.py +++ b/colossalai/communication/utils.py @@ -1,10 +1,11 @@ +from typing import List, Tuple, Union + import torch import torch.distributed as dist from colossalai.context.parallel_mode import ParallelMode from colossalai.core import global_context as gpc from colossalai.utils import get_current_device -from typing import Union, List, Tuple TensorShape = Union[torch.Size, List[int], Tuple[int]]