[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

colossalai/pipeline/schedule/interleaved_pp.py

@@ -6,7 +6,7 @@
 from torch.nn import Module, ModuleList
 from torch.utils._pytree import tree_map
 
-from colossalai.accelerator import get_accelerator, BaseAccelerator
+from colossalai.accelerator import BaseAccelerator, get_accelerator
 from colossalai.interface import OptimizerWrapper
 from colossalai.pipeline.p2p import PipelineP2PCommunication, create_send_metadata
 from colossalai.pipeline.stage_manager import PipelineStageManager

colossalai/shardformer/layer/normalization.py

@@ -17,6 +17,7 @@
 SUPPORT_NPU = False
 try:
     import torch_npu
+
     SUPPORT_NPU = True
 except Exception:
     pass

extensions/pybind/flash_attention/flash_attention_npu.py

@@ -1,6 +1,7 @@
-from ...base_extension import _Extension
 import math
 
+from ...base_extension import _Extension
+
 
 class FlashAttentionNpuExtension(_Extension):
     def __init__(self):