From f566c9b98d3ce435c86789382b2d52891ef4d2ed Mon Sep 17 00:00:00 2001
From: Zirui Zhu <zhuzr21@gmail.com>
Date: Thu, 8 Sep 2022 16:25:57 +0800
Subject: [PATCH] [NFC] polish colossalai/pipeline/utils.py code style (#1562)

---
 colossalai/pipeline/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/pipeline/utils.py b/colossalai/pipeline/utils.py
index d4e759a63..5afed0225 100644
--- 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
-