[CodeStyle][PLR0402] import a.b to from a import b (#52125)

paddle/infrt/tests/models/efficientnet-b4/net/efficientnet.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 
 from .utils import (
     drop_connect,

paddle/infrt/tests/models/efficientnet-b4/net/utils.py

@@ -18,8 +18,8 @@ import re
 from functools import partial
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 
 # Parameters for the entire model (stem, all blocks, and head)
 GlobalParams = collections.namedtuple(

paddle/infrt/tests/models/linear.py

@@ -16,8 +16,8 @@
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.optimizer as opt
+from paddle import nn
 
 BATCH_SIZE = 16
 BATCH_NUM = 4

pyproject.toml

@@ -115,6 +115,9 @@ select = [
     # "B030",
     "B032",
     # "B904",
+
+    # Pylint
+    "PLR0402",
 ]
 unfixable = [
     "NPY001"

python/paddle/audio/features/layers.py

@@ -15,8 +15,7 @@ from functools import partial
 from typing import Optional, Union
 
 import paddle
-import paddle.nn as nn
-from paddle import Tensor
+from paddle import Tensor, nn
 
 from ..functional import compute_fbank_matrix, create_dct, power_to_db
 from ..functional.window import get_window

python/paddle/common_ops_import.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import paddle.fluid as fluid  # noqa: F401
+from paddle import fluid  # noqa: F401
 from paddle.fluid import core, dygraph_utils  # noqa: F401
 from paddle.fluid.core import VarDesc  # noqa: F401
 from paddle.fluid.data_feeder import (  # noqa: F401

python/paddle/cost_model/cost_model.py

@@ -18,7 +18,7 @@ import os
 import numpy as np
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.fluid import core
 
 

python/paddle/dataset/cifar.py

@@ -33,7 +33,7 @@ import tarfile
 import numpy
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/conll05.py

@@ -24,7 +24,7 @@ import gzip
 import tarfile
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/flowers.py

@@ -33,10 +33,9 @@ import functools
 import tarfile
 from multiprocessing import cpu_count
 
-import paddle.utils.deprecated as deprecated
 from paddle.dataset.image import load_image_bytes, simple_transform
 from paddle.reader import map_readers, xmap_readers
-from paddle.utils import try_import
+from paddle.utils import deprecated, try_import
 
 from .common import download
 

python/paddle/dataset/imdb.py

@@ -26,7 +26,7 @@ import string
 import tarfile
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/imikolov.py

@@ -23,7 +23,7 @@ import collections
 import tarfile
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/mnist.py

@@ -24,7 +24,7 @@ import struct
 import numpy
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/movielens.py

@@ -29,7 +29,7 @@ import zipfile
 import numpy as np
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/uci_housing.py

@@ -26,7 +26,7 @@ import tempfile
 import numpy as np
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/voc2012.py

@@ -25,8 +25,8 @@ import tarfile
 import numpy as np
 from PIL import Image
 
-import paddle.utils.deprecated as deprecated
 from paddle.dataset.common import download
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/wmt14.py

@@ -23,7 +23,7 @@ parse training set and test set into paddle reader creators.
 import tarfile
 
 import paddle.dataset.common
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/dataset/wmt16.py

@@ -33,7 +33,7 @@ import tarfile
 from collections import defaultdict
 
 import paddle
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 
 __all__ = []
 

python/paddle/distributed/auto_parallel/engine.py

@@ -23,8 +23,7 @@ import numpy as np
 
 import paddle
 import paddle.distributed.auto_parallel.utils as auto_utils
-import paddle.utils as utils
-from paddle import static
+from paddle import static, utils
 from paddle.distributed import fleet
 from paddle.fluid.executor import _to_name_str
 from paddle.framework import IrGraph

python/paddle/distributed/collective.py

@@ -18,7 +18,7 @@ import os
 import paddle
 
 # (TODO: GhostScreaming) It will be removed later.
-import paddle.fluid.core as core
+from paddle.fluid import core
 from paddle.framework import in_dygraph_mode
 
 from .communication.group import Group, _add_new_group, is_initialized

python/paddle/distributed/communication/all_gather.py

@@ -15,8 +15,8 @@
 import numpy as np
 
 import paddle
-import paddle.distributed.communication.stream as stream
-import paddle.framework as framework
+from paddle import framework
+from paddle.distributed.communication import stream
 
 from .serialization_utils import (
     convert_object_to_tensor,

python/paddle/distributed/communication/all_reduce.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.communication.stream as stream
+from paddle.distributed.communication import stream
 from paddle.distributed.communication.reduce import ReduceOp
 
 

python/paddle/distributed/communication/all_to_all.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.communication.stream as stream
+from paddle.distributed.communication import stream
 
 
 def alltoall(in_tensor_list, out_tensor_list, group=None, sync_op=True):

python/paddle/distributed/communication/batch_isend_irecv.py

@@ -15,7 +15,7 @@
 import contextlib
 
 import paddle.distributed as dist
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _warn_cur_rank_not_in_group,

python/paddle/distributed/communication/broadcast.py

@@ -14,8 +14,8 @@
 
 import paddle
 import paddle.distributed as dist
-import paddle.distributed.communication.stream as stream
-import paddle.framework as framework
+from paddle import framework
+from paddle.distributed.communication import stream
 
 from .serialization_utils import (
     convert_object_to_tensor,

python/paddle/distributed/communication/group.py

@@ -16,7 +16,7 @@ import warnings
 
 import paddle
 import paddle.distributed as dist
-import paddle.framework as framework
+from paddle import framework
 
 
 class Group:

python/paddle/distributed/communication/recv.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.communication.stream as stream
+from paddle.distributed.communication import stream
 
 
 def recv(tensor, src=0, group=None, sync_op=True):

python/paddle/distributed/communication/reduce.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import paddle
-import paddle.distributed.communication.stream as stream
-import paddle.framework as framework
+from paddle import framework
+from paddle.distributed.communication import stream
 
 
 class ReduceOp:

python/paddle/distributed/communication/reduce_scatter.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.communication.stream as stream
+from paddle.distributed.communication import stream
 from paddle.distributed.communication.reduce import ReduceOp
 from paddle.distributed.communication.stream.reduce_scatter import (
     _reduce_scatter_base as _reduce_scatter_base_stream,

python/paddle/distributed/communication/scatter.py

@@ -16,8 +16,8 @@ import numpy as np
 
 import paddle
 import paddle.distributed as dist
-import paddle.distributed.communication.stream as stream
-import paddle.framework as framework
+from paddle import framework
+from paddle.distributed.communication import stream
 
 from .serialization_utils import (
     convert_object_to_tensor,

python/paddle/distributed/communication/send.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.distributed.communication.stream as stream
+from paddle.distributed.communication import stream
 
 
 def send(tensor, dst=0, group=None, sync_op=True):

python/paddle/distributed/communication/stream/all_gather.py

@@ -14,9 +14,9 @@
 
 import paddle
 import paddle.distributed as dist
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import _get_global_group
+from paddle.fluid import data_feeder
 
 
 def _all_gather_into_tensor_in_dygraph(

python/paddle/distributed/communication/stream/all_reduce.py

@@ -12,13 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _warn_cur_rank_not_in_group,
 )
 from paddle.distributed.communication.reduce import ReduceOp, _get_reduce_op
+from paddle.fluid import data_feeder
 
 
 def _all_reduce_in_dygraph(tensor, op, group, sync_op, use_calc_stream):

python/paddle/distributed/communication/stream/all_to_all.py

@@ -14,12 +14,12 @@
 
 import paddle
 import paddle.distributed as dist
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _warn_cur_rank_not_in_group,
 )
+from paddle.fluid import data_feeder
 
 
 def _all_to_all_tensor_in_dygraph(

python/paddle/distributed/communication/stream/broadcast.py

@@ -12,13 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _get_or_throw_group_rank,
     _warn_cur_rank_not_in_group,
 )
+from paddle.fluid import data_feeder
 
 
 def _broadcast_in_dygraph(

python/paddle/distributed/communication/stream/recv.py

@@ -12,13 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _get_or_throw_group_rank,
     _warn_cur_rank_not_in_group,
 )
+from paddle.fluid import data_feeder
 
 
 def _recv_in_dygraph(

python/paddle/distributed/communication/stream/reduce.py

@@ -12,14 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _get_or_throw_group_rank,
     _warn_cur_rank_not_in_group,
 )
 from paddle.distributed.communication.reduce import ReduceOp, _get_reduce_op
+from paddle.fluid import data_feeder
 
 
 def _reduce_in_dygraph(

python/paddle/distributed/communication/stream/reduce_scatter.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import paddle
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _warn_cur_rank_not_in_group,

python/paddle/distributed/communication/stream/scatter.py

@@ -16,13 +16,13 @@ import warnings
 
 import paddle
 import paddle.distributed as dist
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _get_or_throw_group_rank,
     _warn_cur_rank_not_in_group,
 )
+from paddle.fluid import data_feeder
 
 
 def _scatter_tensor_in_dygraph(

python/paddle/distributed/communication/stream/send.py

@@ -12,13 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.fluid.data_feeder as data_feeder
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.communication.group import (
     _get_global_group,
     _get_or_throw_group_rank,
     _warn_cur_rank_not_in_group,
 )
+from paddle.fluid import data_feeder
 
 
 def _send_in_dygraph(

python/paddle/distributed/fleet/base/distributed_strategy.py

@@ -24,9 +24,9 @@ from paddle.fluid.wrapped_decorator import wrap_decorator
 
 protobuf_version = google.protobuf.__version__
 if protobuf_version >= "4.21.0":
-    import google._upb._message as _message
+    from google._upb import _message
 else:
-    import google.protobuf.pyext._message as _message
+    from google.protobuf.pyext import _message
 
 __all__ = []
 

python/paddle/distributed/fleet/base/role_maker.py

@@ -20,10 +20,10 @@ from multiprocessing import Manager, Process
 import numpy as np
 
 import paddle
-import paddle.fluid.core as core
 from paddle.distributed.fleet.base.private_helper_function import (
     wait_server_ready,
 )
+from paddle.fluid import core
 
 __all__ = []
 

python/paddle/distributed/fleet/base/util_factory.py

@@ -24,7 +24,7 @@ import numpy as np
 from google.protobuf import text_format
 
 import paddle
-import paddle.framework as framework
+from paddle import framework
 from paddle.fluid import core, debugger
 from paddle.fluid.proto import framework_pb2
 from paddle.static import Program

python/paddle/distributed/fleet/dataset/dataset.py

@@ -15,7 +15,7 @@
 
 from google.protobuf import text_format
 
-import paddle.fluid.core as core
+from paddle.fluid import core
 from paddle.fluid.proto import data_feed_pb2
 
 __all__ = []

python/paddle/distributed/fleet/fleet.py

@@ -267,7 +267,7 @@ class Fleet:
                 )
         self._role_maker._generate_role()
 
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         fleet.util._set_role_maker(self._role_maker)
 
@@ -1557,7 +1557,7 @@ class Fleet:
             if self._runtime_handle is None:
                 self._runtime_handle = RuntimeFactory()._create_runtime(context)
 
-            import paddle.distributed.fleet as fleet
+            from paddle.distributed import fleet
 
             fleet.util._set_strategy(context["valid_strategy"])
 
@@ -1651,7 +1651,7 @@ class Fleet:
         if self._runtime_handle is None:
             self._runtime_handle = RuntimeFactory()._create_runtime(context)
 
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         fleet.util._set_strategy(context["valid_strategy"])
 

python/paddle/distributed/fleet/launch.py

@@ -63,7 +63,7 @@ import tempfile
 import time
 from argparse import REMAINDER, ArgumentParser
 
-import paddle.framework as framework
+from paddle import framework
 from paddle.distributed.fleet import ascend_utils, cloud_utils, launch_utils
 from paddle.distributed.fleet.elastic import enable_elastic, launch_elastic
 from paddle.distributed.fleet.launch_utils import (

python/paddle/distributed/fleet/launch_utils.py

@@ -28,8 +28,8 @@ import time
 from contextlib import closing
 from distutils.util import strtobool
 
-import paddle.framework as framework
 import paddle.utils.cpp_extension.extension_utils as utils
+from paddle import framework
 
 logger = logging.getLogger("root")
 logger.propagate = False

python/paddle/distributed/fleet/layers/mpu/random.py

@@ -87,7 +87,7 @@ def get_rng_state_tracker():
 
 
 def model_parallel_random_seed(seed=None):
-    import paddle.distributed.fleet as fleet
+    from paddle.distributed import fleet
 
     hcg = fleet.get_hybrid_communicate_group()
     rank = hcg.get_model_parallel_rank()

python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_optimizer.py

@@ -16,8 +16,8 @@ from collections import namedtuple
 
 import hccl.manage.api as hccl
 
-import paddle.framework.core as core
 from paddle.distributed import fleet
+from paddle.framework import core
 from paddle.optimizer import Optimizer
 
 from . import ascend_parser

python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py

@@ -15,7 +15,7 @@ from functools import reduce
 
 import numpy as np
 
-import paddle.framework.core as core
+from paddle.framework import core
 
 __all__ = []
 

python/paddle/distributed/fleet/meta_optimizers/raw_program_optimizer.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 
-import paddle.static as static
+from paddle import static
 from paddle.fluid import core
 from paddle.utils import unique_name
 

python/paddle/distributed/fleet/meta_optimizers/tensor_parallel_optimizer.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 
-import paddle.static as static
+from paddle import static
 
 from .common import (
     OP_ROLE_KEY,

python/paddle/distributed/fleet/meta_parallel/meta_parallel_base.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.nn as nn
+from paddle import nn
 
 __all__ = []
 

python/paddle/distributed/fleet/meta_parallel/parallel_layers/pp_layers.py

@@ -45,8 +45,7 @@ import re
 from functools import partial
 
 import paddle
-import paddle.framework as framework
-import paddle.nn as nn
+from paddle import framework, nn
 from paddle.incubate.distributed.fleet import recompute_hybrid
 
 from ...utils.log_util import layer_to_str, logger

python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 
 import paddle
-import paddle.framework as framework
+from paddle import framework
 
 from ..meta_optimizers.dygraph_optimizer import HybridParallelOptimizer
 from ..utils.hybrid_parallel_util import (

python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py

@@ -15,7 +15,7 @@
 import numpy as np
 
 import paddle
-import paddle.framework as framework
+from paddle import framework
 
 from ...utils.log_util import logger
 from .utils import number_2_dtype, paddle_2_number

python/paddle/distributed/fleet/runtime/the_one_ps.py

@@ -15,7 +15,7 @@
 import os
 import warnings
 
-import paddle.fluid as fluid
+from paddle import fluid
 from paddle.fluid import core
 from paddle.fluid.compiler import CompiledProgram
 from paddle.fluid.executor import Executor
@@ -820,7 +820,7 @@ class TheOnePSRuntime(RuntimeBase):
             send_ctx, dense_map, proto_txt, string_hosts, fluid.global_scope()
         )
 
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         fleet.util.barrier()
         info = self._communicator.get_client_info()
@@ -1566,7 +1566,7 @@ class TheOnePSRuntime(RuntimeBase):
             )
         else:
             threshold = 0
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         fleet.util.barrier()
         if self.role_maker._is_first_worker():

python/paddle/distributed/fleet/utils/__init__.py

@@ -15,7 +15,7 @@
 from .fs import LocalFS  # noqa: F401
 from .fs import HDFSClient  # noqa: F401
 from .ps_util import DistributedInfer  # noqa: F401
-import paddle.utils.deprecated as deprecated
+from paddle.utils import deprecated
 from paddle.distributed import fleet
 
 import paddle

python/paddle/distributed/fleet/utils/hybrid_parallel_inference.py

@@ -16,10 +16,10 @@ from collections import defaultdict
 
 import numpy as np
 
-import paddle.distributed.fleet as fleet
+from paddle.distributed import fleet
 
 # (TODO: GhostScreaming) It will be removed later.
-import paddle.fluid.core as core
+from paddle.fluid import core
 from paddle.fluid.framework import in_dygraph_mode
 from paddle.framework import Block, Program
 

python/paddle/distributed/fleet/utils/ps_util.py

@@ -45,7 +45,7 @@ class DistributedInfer:
     def init_distributed_infer_env(
         self, exe, loss, role_maker=None, dirname=None
     ):
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         if fleet.fleet._runtime_handle is None:
             fleet.init(role_maker=role_maker)
@@ -73,7 +73,7 @@ class DistributedInfer:
             global_main_program = self.origin_main_program
 
     def _get_sparse_table_map(self):
-        import paddle.distributed.fleet as fleet
+        from paddle.distributed import fleet
 
         if self.sparse_table_maps is None:
             self.sparse_table_maps = {}

python/paddle/distributed/passes/auto_parallel_fp16.py

@@ -764,7 +764,7 @@ class FP16Pass(AMPPass):
             __target_dtype = core.VarDesc.VarType.FP16
 
         elif self.target_dtype == "bfloat16":
-            import paddle.static.amp.bf16.amp_utils as amp_utils
+            from paddle.static.amp.bf16 import amp_utils
 
             AMPList = amp_utils.AutoMixedPrecisionListsBF16
             __target_dtype = core.VarDesc.VarType.BF16

python/paddle/distributed/passes/ps_trainer_pass.py

@@ -17,8 +17,8 @@ import os
 from _collections import defaultdict
 
 import paddle
-import paddle.fluid.framework as framework
 from paddle.distributed.passes.pass_base import PassBase, register_pass
+from paddle.fluid import framework
 from paddle.framework import core
 from paddle.static import Parameter, Program
 

python/paddle/distributed/ps/utils/ps_program_builder.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import paddle
-import paddle.fluid as fluid
+from paddle import fluid
 from paddle.distributed.fleet.base.private_helper_function import (
     wait_server_ready,
 )

python/paddle/distributed/rpc/rpc.py

@@ -18,10 +18,10 @@ import pickle
 import time
 from collections import namedtuple
 
-import paddle.fluid.core as core
 from paddle.distributed.launch.context import Node
 from paddle.distributed.rpc.internal import PythonFunc, _serialize
 from paddle.distributed.utils.launch_utils import logger
+from paddle.fluid import core
 
 WorkerInfo = namedtuple("WorkerInfo", ["name", "rank", "ip", "port"])
 

python/paddle/fluid/tests/cpp_extension/test_cpp_extension_setup.py

@@ -20,7 +20,7 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 
 

python/paddle/fluid/tests/custom_op/custom_raw_op_kernel_op_setup.py

@@ -17,7 +17,7 @@ import os
 from utils import extra_compile_args, paddle_includes
 
 import paddle
-import paddle.fluid.core as core
+from paddle.fluid import core
 from paddle.utils.cpp_extension import CppExtension, CUDAExtension, setup
 
 if paddle.is_compiled_with_cuda():

python/paddle/fluid/tests/custom_op/test_custom_concat.py

@@ -19,7 +19,7 @@ import numpy as np
 from utils import extra_cc_args, extra_nvcc_args, paddle_includes
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_op/test_custom_conj.py

@@ -19,7 +19,7 @@ import numpy as np
 from utils import extra_cc_args, extra_nvcc_args, paddle_includes
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_op/test_custom_inplace.py

@@ -19,7 +19,7 @@ import numpy as np
 from utils import extra_cc_args, extra_nvcc_args, paddle_includes
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_op/test_custom_linear.py

@@ -20,7 +20,7 @@ from utils import extra_cc_args, extra_nvcc_args, paddle_includes
 
 import paddle
 import paddle.nn.functional as F
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_op/test_custom_optional.py

@@ -19,7 +19,7 @@ import numpy as np
 from utils import extra_cc_args, extra_nvcc_args, paddle_includes
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_op/test_custom_relu_op_setup.py

@@ -20,7 +20,7 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 from paddle.vision.transforms import Compose, Normalize
 

python/paddle/fluid/tests/custom_op/test_custom_relu_op_xpu_setup.py

@@ -20,7 +20,7 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 from paddle.vision.transforms import Compose, Normalize
 

python/paddle/fluid/tests/custom_op/test_custom_tensor_operator.py

@@ -19,7 +19,7 @@ import numpy as np
 from utils import extra_cc_args, paddle_includes
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.utils.cpp_extension import get_build_directory, load
 from paddle.utils.cpp_extension.extension_utils import run_cmd
 

python/paddle/fluid/tests/custom_runtime/test_custom_cpu_profiler_plugin.py

@@ -49,7 +49,7 @@ class TestCustomCPUProfilerPlugin(unittest.TestCase):
 
     def test_custom_profiler(self):
         import paddle
-        import paddle.profiler as profiler
+        from paddle import profiler
 
         paddle.set_device('custom_cpu')
 

python/paddle/fluid/tests/custom_runtime/test_custom_op_setup.py

@@ -44,7 +44,7 @@ def custom_relu_dynamic(func, device, dtype, np_x, use_func=True):
 
 def custom_relu_static(func, device, dtype, np_x, use_func=True):
     import paddle
-    import paddle.static as static
+    from paddle import static
 
     paddle.enable_static()
     paddle.set_device(device)

python/paddle/fluid/tests/unittests/ascend_group.py

@@ -16,11 +16,10 @@ import os
 from collections import namedtuple
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
 from paddle.distributed import fleet
 from paddle.distributed.fleet.meta_optimizers.ascend import ascend_optimizer
-from paddle.fluid import unique_name
+from paddle.fluid import core, unique_name
 from paddle.fluid.layer_helper import LayerHelper
 
 Block = namedtuple('Block', ['program'])

python/paddle/fluid/tests/unittests/asp/asp_pruning_base.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/asp/test_asp_customized_pruning.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
+from paddle.fluid import core
 from paddle.incubate import asp as sparsity
 from paddle.incubate.asp.supported_layer_list import (
     supported_layers_and_prune_func_map,

python/paddle/fluid/tests/unittests/asp/test_asp_optimize_dynamic.py

@@ -18,7 +18,7 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid.core as core
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 

python/paddle/fluid/tests/unittests/asp/test_asp_optimize_static.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/asp/test_asp_pruning_static.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/asp/test_asp_save_load.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
-import paddle.fluid.core as core
+from paddle import fluid
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 

python/paddle/fluid/tests/unittests/asp/test_fleet_with_asp_dynamic.py

@@ -19,8 +19,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.distributed.fleet as fleet
-import paddle.fluid.core as core
+from paddle.distributed import fleet
+from paddle.fluid import core
 from paddle.incubate.asp import ASPHelper
 
 cuda_visible_devices = os.getenv('CUDA_VISIBLE_DEVICES')

python/paddle/fluid/tests/unittests/asp/test_fleet_with_asp_sharding.py

@@ -19,8 +19,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.distributed.fleet as fleet
-import paddle.fluid as fluid
+from paddle import fluid
+from paddle.distributed import fleet
 from paddle.incubate import asp as sparsity
 from paddle.incubate.asp import ASPHelper
 

python/paddle/fluid/tests/unittests/asp/test_fleet_with_asp_static.py

@@ -19,8 +19,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.distributed.fleet as fleet
-import paddle.fluid as fluid
+from paddle import fluid
+from paddle.distributed import fleet
 from paddle.incubate import asp as sparsity
 from paddle.incubate.asp import ASPHelper
 

python/paddle/fluid/tests/unittests/auto_checkpoint_utils.py

@@ -18,8 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.fluid as fluid
 import paddle.fluid.incubate.checkpoint.auto_checkpoint as acp
+from paddle import fluid
 from paddle.fluid import unique_name
 from paddle.fluid.framework import program_guard
 

python/paddle/fluid/tests/unittests/auto_parallel/auto_parallel_relaunch_model.py

@@ -15,10 +15,8 @@
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
-import paddle.utils as utils
+from paddle import nn, static, utils
 from paddle.distributed import fleet
 from paddle.distributed.fleet import auto
 

python/paddle/fluid/tests/unittests/auto_parallel/auto_parallel_relaunch_with_gpt_planner.py

@@ -17,7 +17,7 @@ import sys
 import numpy as np
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.distributed import fleet
 
 sys.path.append("..")

python/paddle/fluid/tests/unittests/auto_parallel/auto_parallel_relaunch_with_planner.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import paddle
-import paddle.static as static
+from paddle import static
 from paddle.distributed import fleet
 from paddle.distributed.auto_parallel.cluster import Cluster
 from paddle.distributed.auto_parallel.cost import CostEstimator

python/paddle/fluid/tests/unittests/auto_parallel/engine_api.py

@@ -18,10 +18,8 @@ import tempfile
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
-import paddle.utils as utils
+from paddle import nn, static, utils
 from paddle.distributed.fleet import auto
 from paddle.io import Dataset
 

python/paddle/fluid/tests/unittests/auto_parallel/engine_api_dp.py

@@ -18,8 +18,8 @@ import tempfile
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 from paddle.distributed.fleet import auto
 from paddle.io import Dataset
 

python/paddle/fluid/tests/unittests/auto_parallel/iterable_dataset.py

@@ -18,8 +18,8 @@ import tempfile
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 from paddle.distributed.fleet import auto
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/auto_parallel/optimization_tuner_api.py

@@ -15,8 +15,8 @@
 from engine_api_dp import MyDataset
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 from paddle.distributed.fleet import auto
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/auto_parallel/test_base_cost.py

@@ -20,10 +20,8 @@ import unittest
 from test_cluster import cluster_json
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
-import paddle.utils as utils
+from paddle import nn, static, utils
 from paddle.distributed import fleet
 from paddle.distributed.auto_parallel.cluster import Cluster
 from paddle.distributed.auto_parallel.completion import Completer

python/paddle/fluid/tests/unittests/auto_parallel/test_conditional_block_reshard.py

@@ -15,8 +15,8 @@
 import unittest
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 from paddle.distributed.fleet import auto
 from paddle.static import InputSpec
 

python/paddle/fluid/tests/unittests/auto_parallel/test_dist_attr_v2.py

@@ -18,9 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
+from paddle import nn, static
 from paddle.distributed import fleet
 from paddle.distributed.auto_parallel.dist_context import (
     DistributedContext,

python/paddle/fluid/tests/unittests/auto_parallel/test_dist_context.py

@@ -18,9 +18,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
+from paddle import nn, static
 from paddle.distributed import fleet
 from paddle.distributed.auto_parallel.dist_context import DistributedContext
 from paddle.distributed.fleet import auto

python/paddle/fluid/tests/unittests/auto_parallel/test_dist_saver.py

@@ -17,8 +17,8 @@ import tempfile
 import unittest
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 from paddle.distributed.fleet import auto
 
 paddle.enable_static()

python/paddle/fluid/tests/unittests/auto_parallel/test_engine_api_error.py

@@ -17,9 +17,8 @@ import unittest
 import numpy as np
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
-import paddle.static as static
+from paddle import nn, static
 from paddle.distributed.fleet import auto
 from paddle.io import Dataset