[NFC] polish...

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/where_handler.py code style (#2723)

[NFC] polish...
[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/where_handler.py code style (#2723)
51c45c24 · yuxuan-lou · GitHub · 4603538d · 51c45c24
Unverified Commit 51c45c24 authored Feb 15, 2023 by yuxuan-lou Committed by GitHub Feb 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

colossalai/auto_parallel/tensor_shard/deprecated/op_handler/where_handler.py ...allel/tensor_shard/deprecated/op_handler/where_handler.py +6 -4

No files found.
--- a/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/where_handler.py
+++ b/colossalai/auto_parallel/tensor_shard/deprecated/op_handler/where_handler.py
@@ -6,10 +6,12 @@ from typing import Dict, List

 import torch

-from colossalai.auto_parallel.tensor_shard.deprecated._utils import (enumerate_all_possible_1d_sharding,
-                                                                     enumerate_all_possible_2d_sharding,
-                                                                     ignore_sharding_exception)
-from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import (ShardingStrategy, StrategiesVector)
+from colossalai.auto_parallel.tensor_shard.deprecated._utils import (
+    enumerate_all_possible_1d_sharding,
+    enumerate_all_possible_2d_sharding,
+    ignore_sharding_exception,
+)
+from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import ShardingStrategy, StrategiesVector
 from colossalai.tensor.shape_consistency import ShapeConsistencyManager
 from colossalai.tensor.sharding_spec import ShardingSpec