fix format (#585)

Co-authored-by: fanjifu <FAN>

fix format (#585)
Co-authored-by: fanjifu <FAN>
ffad81e1 · fanjinfucool · binmakeswell · 6582aedc · ffad81e1
Commit ffad81e1 authored Mar 31, 2022 by fanjinfucool Committed by binmakeswell Apr 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

colossalai/engine/gradient_handler/__init__.py colossalai/engine/gradient_handler/__init__.py +4 -4

No files found.
--- 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'
+]