diff --git a/python/paddle/distributed/__init__.py b/python/paddle/distributed/__init__.py
index f7db380a2a755d..5f5cca12add0c4 100644
--- a/python/paddle/distributed/__init__.py
+++ b/python/paddle/distributed/__init__.py
@@ -84,7 +84,6 @@
     dtensor_from_fn,
     reshard,
     shard_layer,
-    shard_optimizer,
     ShardOptimizer,
 )
 
@@ -159,6 +158,5 @@
     "Shard",
     "Replicate",
     "Partial",
-    "shard_optimizer",
     "ShardOptimizer",
 ]