[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py code style (#2720)

Co-authored-by: Fuzhao Xue <fuzhao@login2.ls6.tacc.utexas.edu>

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py code style (#2720)
Co-authored-by: Fuzhao Xue <fuzhao@login2.ls6.tacc.utexas.edu>
e81caeb4 · Xue Fuzhao · GitHub · 51c45c24 · e81caeb4
Unverified Commit e81caeb4 authored Feb 15, 2023 by Xue Fuzhao Committed by GitHub Feb 15, 2023
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 6 deletions

colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py ...salai/auto_parallel/tensor_shard/deprecated/cost_graph.py +8 -6

No files found.
--- a/colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py
+++ b/colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py
-from typing import List
 import math
+from typing import List
+
 from torch.fx.node import Node
+
 from .constants import INFINITY_COST