From 64350029fecb167db415c82ad41cbd8190dde9e5 Mon Sep 17 00:00:00 2001
From: Xu Kai <xukai16@foxmail.com>
Date: Wed, 29 Mar 2023 02:45:11 -0500
Subject: [PATCH] [NFC] polish colossalai/gemini/paramhooks/_param_hookmgr.py
 code style

---
 colossalai/gemini/paramhooks/_param_hookmgr.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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):