diff --git a/colossalai/nn/layer/colossalai_layer/dropout.py b/colossalai/nn/layer/colossalai_layer/dropout.py
index 0df6698d5..cc2d9a0a7 100644
--- a/colossalai/nn/layer/colossalai_layer/dropout.py
+++ b/colossalai/nn/layer/colossalai_layer/dropout.py
@@ -13,6 +13,7 @@ class Dropout(ColossalaiModule):
         p (float, optional): probability of an element to be zeroed, defaults 0.5.
         inplace (bool, optional): whether to do dropout in-place, default to be False.
     """
+
     def __init__(self, p: float = 0.5, inplace: bool = False) -> None:
         tensor_parallel = get_tensor_parallel_mode()
         if tensor_parallel == "1d":