[NFC] polish <colossalai/auto_parallel/tensor_shard/deprecated/constants.py> code style (#2339)

2 years ago · bd12a49e2a
parent 35427bcab4
commit bd12a49e2a
1 changed files with 2 additions and 1 deletions
--- a/colossalai/auto_parallel/tensor_shard/deprecated/constants.py
+++ b/colossalai/auto_parallel/tensor_shard/deprecated/constants.py
@ -1,6 +1,7 @@
-import torch
 import operator

+import torch
+
 __all__ = [
    'ELEMENTWISE_MODULE_OP', 'ELEMENTWISE_FUNC_OP', 'RESHAPE_FUNC_OP', 'CONV_MODULE_OP', 'CONV_FUNC_OP',
    'LINEAR_MODULE_OP', 'LINEAR_FUNC_OP', 'BATCHNORM_MODULE_OP', 'POOL_MODULE_OP', 'NON_PARAM_FUNC_OP', 'BCAST_FUNC_OP',