[NFC] polish colossalai/gemini/paramhooks/_param_hookmgr.py code style

2023-03-29 02:45:11 -05:00 · 2023-03-29 02:45:11 -05:00 · 64350029fe
parent 1ce9d0c531
commit 64350029fe
1 changed files with 3 additions and 2 deletions
--- 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):