diff --git a/pyproject.toml b/pyproject.toml index b5165bdb5bea33..912b98c0bb85d3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -135,6 +135,3 @@ known-first-party = ["paddle"] "python/paddle/distributed/launch/context/__init__.py" = ["I001"] "python/paddle/distributed/launch/controllers/__init__.py" = ["I001"] "python/paddle/distributed/passes/__init__.py" = ["I001"] -"python/paddle/distributed/rpc/__init__.py" = ["I001"] -"python/paddle/incubate/distributed/fleet/__init__.py" = ["I001"] -"python/paddle/incubate/distributed/fleet/parameter_server/pslib/__init__.py" = ["I001"] diff --git a/python/paddle/distributed/rpc/__init__.py b/python/paddle/distributed/rpc/__init__.py index 692600a93b9297..537416fd45e157 100644 --- a/python/paddle/distributed/rpc/__init__.py +++ b/python/paddle/distributed/rpc/__init__.py @@ -13,13 +13,13 @@ # limitations under the License. from paddle.distributed.rpc.rpc import ( + get_all_worker_infos, + get_current_worker_info, + get_worker_info, init_rpc, - shutdown, rpc_async, rpc_sync, - get_worker_info, - get_all_worker_infos, - get_current_worker_info, + shutdown, ) __all__ = [ diff --git a/python/paddle/incubate/distributed/fleet/__init__.py b/python/paddle/incubate/distributed/fleet/__init__.py index f098e4c951c0e7..fad13d1d71d74b 100644 --- a/python/paddle/incubate/distributed/fleet/__init__.py +++ b/python/paddle/incubate/distributed/fleet/__init__.py @@ -13,8 +13,8 @@ # limitations under the License. from paddle.distributed.fleet.recompute import ( - recompute_sequential, recompute_hybrid, + recompute_sequential, ) __all__ = ["recompute_sequential", "recompute_hybrid"] diff --git a/python/paddle/incubate/distributed/fleet/parameter_server/pslib/__init__.py b/python/paddle/incubate/distributed/fleet/parameter_server/pslib/__init__.py index b5cd746e35890d..c53d9a87f35cb4 100644 --- a/python/paddle/incubate/distributed/fleet/parameter_server/pslib/__init__.py +++ b/python/paddle/incubate/distributed/fleet/parameter_server/pslib/__init__.py @@ -30,8 +30,10 @@ MPISymetricRoleMaker, ) -from .optimizer_factory import DistributedAdam # noqa: F401 -from .optimizer_factory import FLEET_GLOBAL_DICT +from .optimizer_factory import ( + FLEET_GLOBAL_DICT, + DistributedAdam, # noqa: F401 +) class PSLib(Fleet):