From ffad81e1d1ce7b450227ec5c5da166361d935ad7 Mon Sep 17 00:00:00 2001 From: fanjinfucool <56330779+fanjinfucool@users.noreply.github.com> Date: Thu, 31 Mar 2022 20:36:09 +0800 Subject: [PATCH] fix format (#585) Co-authored-by: fanjifu --- colossalai/engine/gradient_handler/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/colossalai/engine/gradient_handler/__init__.py b/colossalai/engine/gradient_handler/__init__.py index b6503b778..6177da69b 100644 --- a/colossalai/engine/gradient_handler/__init__.py +++ b/colossalai/engine/gradient_handler/__init__.py @@ -6,7 +6,7 @@ from ._pipeline_parallel_gradient_handler import PipelineSharedModuleGradientHan from ._moe_gradient_handler import MoeGradientHandler from ._sequence_parallel_gradient_handler import SequenceParallelGradientHandler - -__all__ = ['BaseGradientHandler', 'DataParallelGradientHandler', - 'ZeROGradientHandler', 'PipelineSharedModuleGradientHandler', - 'MoeGradientHandler', 'SequenceParallelGradientHandler'] \ No newline at end of file +__all__ = [ + 'BaseGradientHandler', 'DataParallelGradientHandler', 'ZeROGradientHandler', 'PipelineSharedModuleGradientHandler', + 'MoeGradientHandler', 'SequenceParallelGradientHandler' +]