[NFC] polish runtime_preparation_pass style (#4266)

2023-07-18 11:54:09 +08:00 · 2023-07-18 11:54:09 +08:00 · fee553288b
parent 3883db452c
commit fee553288b
1 changed files with 1 additions and 1 deletions
--- a/colossalai/auto_parallel/passes/runtime_preparation_pass.py
+++ b/colossalai/auto_parallel/passes/runtime_preparation_pass.py
@ -55,7 +55,7 @@ def size_processing(size: Union[int, torch.Size],


 def solution_annotation_pass(gm: torch.fx.GraphModule, solution: List[int],
-                               strategies_constructor: StrategiesConstructor):
+                             strategies_constructor: StrategiesConstructor):
    """
    This method is used to stick the solution strategy to the nodes and add the information
    required in runtime into graph as placeholder nodes.