[NFC] polish colossalai/pipeline/utils.py code style (#1562)

f566c9b9 · Zirui Zhu · Frank Lee · e070ca45 · f566c9b9
Commit f566c9b9 authored Sep 08, 2022 by Zirui Zhu Committed by Frank Lee Sep 08, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/pipeline/utils.py colossalai/pipeline/utils.py +1 -1

No files found.
--- a/colossalai/pipeline/utils.py
+++ b/colossalai/pipeline/utils.py
@@ -250,6 +250,7 @@ def call_module(module, args=None, kwargs=None):
    else:
        return module(*args_needed, **kwargs)

+
 def customized_partition(exec_seq):
    '''
    This function will analyze the exec_seq. In the exec_seq, users will use 'SPLIT_NODE' as an 
@@ -269,4 +270,3 @@ def customized_partition(exec_seq):
                stop += 1
    customized_parts[rank] = [(start, stop)]
    return customized_parts
-