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

This commit is contained in:
shenggan 2022-05-16 12:33:22 +08:00 committed by binmakeswell
parent b67eebd20f
commit 18542b47fc