diff --git a/colossalai/zero/sharded_param/__init__.py b/colossalai/zero/sharded_param/__init__.py
index 98544c381..5642a504a 100644
--- a/colossalai/zero/sharded_param/__init__.py
+++ b/colossalai/zero/sharded_param/__init__.py
@@ -1,5 +1,4 @@
 from colossalai.zero.sharded_param.sharded_tensor import ShardedTensor
 from colossalai.zero.sharded_param.sharded_param import ShardedParamV2
 
-__all__ = [
-    'ShardedTensor', 'ShardedParamV2']
+__all__ = ['ShardedTensor', 'ShardedParamV2']