[NFC] polish colossalai/nn/layer/parallel_2p5d/layers.py code style (#972)

This commit is contained in:
Zirui Zhu 2022-05-16 11:27:33 +08:00 committed by binmakeswell
parent 632e94abde
commit 598cde4a0f