diff --git a/python/paddle/distributed/__init__.py b/python/paddle/distributed/__init__.py index 4934f2093f3fb768fa718364f7018cc185035611..20007f76ed5e411d1bf8a73c0dc074f586a17912 100644 --- a/python/paddle/distributed/__init__.py +++ b/python/paddle/distributed/__init__.py @@ -13,7 +13,7 @@ # limitations under the License. from .spawn import spawn # noqa: F401 -from paddle.distributed.fleet.launch import launch # noqa: F401 +from .fleet.launch import launch # noqa: F401 from .parallel import init_parallel_env # noqa: F401 from .parallel import get_rank # noqa: F401 diff --git a/python/paddle/distributed/fleet/elastic/collective.py b/python/paddle/distributed/fleet/elastic/collective.py index 94fe6a54b5809b2daee9a9335421912bfddaa108..83f0e85db2badfeff4865f974dd2c9ef2053b0a1 100644 --- a/python/paddle/distributed/fleet/elastic/collective.py +++ b/python/paddle/distributed/fleet/elastic/collective.py @@ -13,8 +13,8 @@ # limitations under the License. from paddle.distributed.fleet import launch_utils -import paddle.distributed.fleet.cloud_utils as cloud_utils -import paddle.distributed.fleet.ascend_utils as ascend_utils +from paddle.distributed.fleet import cloud_utils +from paddle.distributed.fleet import ascend_utils from paddle.distributed.fleet.launch_utils import * diff --git a/python/paddle/distributed/fleet/launch.py b/python/paddle/distributed/fleet/launch.py index 9a2c5b551ccb0619c615484f6a747fa1e85f93fa..c0a1c359d17c63cd7660019b8823c73ef478f005 100644 --- a/python/paddle/distributed/fleet/launch.py +++ b/python/paddle/distributed/fleet/launch.py @@ -72,8 +72,8 @@ from paddle.distributed.fleet import launch_utils # TODO(danleifeng): Don't import * from a module from paddle.distributed.fleet.launch_utils import * -import paddle.distributed.fleet.cloud_utils as cloud_utils -import paddle.distributed.fleet.ascend_utils as ascend_utils +from paddle.distributed.fleet import cloud_utils +from paddle.distributed.fleet import ascend_utils from paddle.distributed.fleet.elastic import enable_elastic, launch_elastic