[NFC] polish colossalai/amp/naive_amp/_utils.py code style (#1816)

* [NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714)

* [NFC] polish colossalai/zero/sharded_param/__init__.py code style

* [NFC] polish colossalai/amp/naive_amp/_utils.py code style
Co-authored-by: shenggan <csg19971016@gmail.com>
Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>

colossalai/amp/naive_amp/_utils.py

 from typing import List
+
 from torch import Tensor