diff --git a/colossalai/builder/__init__.py b/colossalai/builder/__init__.py
index 3edb4b6bc..cf09e1e7a 100644
--- a/colossalai/builder/__init__.py
+++ b/colossalai/builder/__init__.py
@@ -1,5 +1,3 @@
 from .builder import build_from_config, build_from_registry, build_gradient_handler
 
-__all__ = [
-    'build_gradient_handler', 'build_from_config', 'build_from_registry'
-]
+__all__ = ['build_gradient_handler', 'build_from_config', 'build_from_registry']