diff --git a/colossalai/gemini/paramhooks/_param_hookmgr.py b/colossalai/gemini/paramhooks/_param_hookmgr.py
index ee57cb46a..84f32be35 100644
--- a/colossalai/gemini/paramhooks/_param_hookmgr.py
+++ b/colossalai/gemini/paramhooks/_param_hookmgr.py
@@ -1,6 +1,7 @@
-from typing import Callable, List
-import torch
 import functools
+from typing import Callable, List
+
+import torch
 
 
 class BaseParamHookMgr(object):