diff --git a/colossalai/auto_parallel/tensor_shard/deprecated/__init__.py b/colossalai/auto_parallel/tensor_shard/deprecated/__init__.py index a081ce69c..bd47f2adf 100644 --- a/colossalai/auto_parallel/tensor_shard/deprecated/__init__.py +++ b/colossalai/auto_parallel/tensor_shard/deprecated/__init__.py @@ -1,6 +1,6 @@ -from .options import SolverOptions -from .strategies_constructor import StrategiesConstructor -from .sharding_strategy import ShardingStrategy, StrategiesVector from .cost_graph import CostGraph +from .graph_analysis import GraphAnalyser +from .options import SolverOptions +from .sharding_strategy import ShardingStrategy, StrategiesVector from .solver import Solver -from .graph_analysis import GraphAnalyser \ No newline at end of file +from .strategies_constructor import StrategiesConstructor