Unify dpnp public API exports by refactoring dpnp_iface.py

dpnp/dpnp_iface.py

@@ -39,7 +39,6 @@
 
 """
 # pylint: disable=protected-access
-# pylint: disable=redefined-outer-name
 
 import os
 
@@ -51,85 +50,15 @@
 from dpctl.tensor._device import normalize_queue_device
 
 import dpnp
-from dpnp.dpnp_algo import *
 from dpnp.dpnp_array import dpnp_array
 
-__all__ = [
-    "are_same_logical_tensors",
-    "asnumpy",
-    "as_usm_ndarray",
-    "check_limitations",
-    "check_supported_arrays_type",
-    "default_float_type",
-    "get_dpnp_descriptor",
-    "get_include",
-    "get_normalized_queue_device",
-    "get_result_array",
-    "get_usm_ndarray",
-    "get_usm_ndarray_or_scalar",
-    "is_cuda_backend",
-    "is_supported_array_or_scalar",
-    "is_supported_array_type",
-    "synchronize_array_data",
-]
-
-from dpnp.dpnp_iface_arraycreation import *
-from dpnp.dpnp_iface_arraycreation import __all__ as __all__arraycreation
-from dpnp.dpnp_iface_bitwise import *
-from dpnp.dpnp_iface_bitwise import __all__ as __all__bitwise
-from dpnp.dpnp_iface_counting import *
-from dpnp.dpnp_iface_counting import __all__ as __all__counting
-from dpnp.dpnp_iface_functional import *
-from dpnp.dpnp_iface_functional import __all__ as __all__functional
-from dpnp.dpnp_iface_histograms import *
-from dpnp.dpnp_iface_histograms import __all__ as __all__histograms
-from dpnp.dpnp_iface_indexing import *
-from dpnp.dpnp_iface_indexing import __all__ as __all__indexing
-from dpnp.dpnp_iface_linearalgebra import *
-from dpnp.dpnp_iface_linearalgebra import __all__ as __all__linearalgebra
-from dpnp.dpnp_iface_logic import *
-from dpnp.dpnp_iface_logic import __all__ as __all__logic
-from dpnp.dpnp_iface_manipulation import *
-from dpnp.dpnp_iface_manipulation import __all__ as __all__manipulation
-from dpnp.dpnp_iface_mathematical import *
-from dpnp.dpnp_iface_mathematical import __all__ as __all__mathematical
-from dpnp.dpnp_iface_nanfunctions import *
-from dpnp.dpnp_iface_nanfunctions import __all__ as __all__nanfunctions
-from dpnp.dpnp_iface_searching import *
-from dpnp.dpnp_iface_searching import __all__ as __all__searching
-from dpnp.dpnp_iface_sorting import *
-from dpnp.dpnp_iface_sorting import __all__ as __all__sorting
-from dpnp.dpnp_iface_statistics import *
-from dpnp.dpnp_iface_statistics import __all__ as __all__statistics
-from dpnp.dpnp_iface_trigonometric import *
-from dpnp.dpnp_iface_trigonometric import __all__ as __all__trigonometric
-from dpnp.dpnp_iface_window import *
-from dpnp.dpnp_iface_window import __all__ as __all__window
-
 # pylint: disable=no-name-in-module
 from .dpnp_utils import (
     dpnp_descriptor,
     map_dtype_to_device,
     use_origin_backend,
 )
 
-__all__ += __all__arraycreation
-__all__ += __all__bitwise
-__all__ += __all__counting
-__all__ += __all__functional
-__all__ += __all__histograms
-__all__ += __all__indexing
-__all__ += __all__linearalgebra
-__all__ += __all__logic
-__all__ += __all__manipulation
-__all__ += __all__mathematical
-__all__ += __all__nanfunctions
-__all__ += __all__searching
-__all__ += __all__sorting
-__all__ += __all__statistics
-__all__ += __all__trigonometric
-__all__ += __all__window
-
 
 def are_same_logical_tensors(ar1, ar2):
     """
@@ -437,7 +366,7 @@ def get_dpnp_descriptor(
     # If input object is a scalar, it means it was allocated on host memory.
     # We need to copy it to USM memory according to compute follows data.
     if dpnp.isscalar(ext_obj):
-        ext_obj = array(
+        ext_obj = dpnp.array(
             ext_obj,
             dtype=alloc_dtype,
             usm_type=alloc_usm_type,
@@ -464,7 +393,7 @@ def get_dpnp_descriptor(
             ext_obj_offset = 0
 
         if ext_obj.strides != shape_offsets or ext_obj_offset != 0:
-            ext_obj = array(ext_obj, order="C")
+            ext_obj = dpnp.array(ext_obj, order="C")
 
     # while dpnp functions are based on DPNP_QUEUE
     # we need to create a copy on device associated with DPNP_QUEUE
@@ -478,7 +407,7 @@ def get_dpnp_descriptor(
             dpctl.utils.get_execution_queue([queue, default_queue]) is not None
         )
         if not queue_is_default:
-            ext_obj = array(ext_obj, sycl_queue=default_queue)
+            ext_obj = dpnp.array(ext_obj, sycl_queue=default_queue)
 
     dpnp_desc = dpnp_descriptor(ext_obj)
     if dpnp_desc.is_valid:  # pylint: disable=using-constant-test

dpnp/dpnp_iface_arraycreation.py

@@ -61,47 +61,6 @@
 # pylint: disable=no-name-in-module
 from .dpnp_utils import get_usm_allocations, map_dtype_to_device
 
-__all__ = [
-    "arange",
-    "array",
-    "asanyarray",
-    "asarray",
-    "ascontiguousarray",
-    "asfortranarray",
-    "astype",
-    "copy",
-    "diag",
-    "diagflat",
-    "empty",
-    "empty_like",
-    "eye",
-    "frombuffer",
-    "fromfile",
-    "fromfunction",
-    "fromiter",
-    "fromstring",
-    "from_dlpack",
-    "full",
-    "full_like",
-    "geomspace",
-    "identity",
-    "linspace",
-    "loadtxt",
-    "logspace",
-    "meshgrid",
-    "mgrid",
-    "ogrid",
-    "ones",
-    "ones_like",
-    "trace",
-    "tri",
-    "tril",
-    "triu",
-    "vander",
-    "zeros",
-    "zeros_like",
-]
-
 
 def _get_empty_array(
     a,

dpnp/dpnp_iface_bitwise.py

@@ -49,21 +49,6 @@
 import dpnp.backend.extensions.ufunc._ufunc_impl as ufi
 from dpnp.dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc
 
-__all__ = [
-    "binary_repr",
-    "bitwise_and",
-    "bitwise_count",
-    "bitwise_invert",
-    "bitwise_left_shift",
-    "bitwise_not",
-    "bitwise_or",
-    "bitwise_right_shift",
-    "bitwise_xor",
-    "invert",
-    "left_shift",
-    "right_shift",
-]
-
 
 def binary_repr(num, width=None):
     """

dpnp/dpnp_iface_counting.py

@@ -43,8 +43,6 @@
 
 import dpnp
 
-__all__ = ["count_nonzero"]
-
 
 def count_nonzero(a, axis=None, *, keepdims=False, out=None):
     """

dpnp/dpnp_iface_functional.py

@@ -51,8 +51,6 @@
 # pylint: disable=no-name-in-module
 from dpnp.dpnp_utils import get_usm_allocations
 
-__all__ = ["apply_along_axis", "apply_over_axes", "piecewise"]
-
 
 def apply_along_axis(func1d, axis, arr, *args, **kwargs):
     """

dpnp/dpnp_iface_histograms.py

@@ -57,15 +57,6 @@
 # pylint: disable=no-name-in-module
 from .dpnp_utils import get_usm_allocations
 
-__all__ = [
-    "bincount",
-    "digitize",
-    "histogram",
-    "histogram_bin_edges",
-    "histogram2d",
-    "histogramdd",
-]
-
 # range is a keyword argument to many functions, so save the builtin so they can
 # use it.
 _range = range

dpnp/dpnp_iface_indexing.py

@@ -64,36 +64,6 @@
 from .dpnp_array import dpnp_array
 from .dpnp_utils import call_origin, get_usm_allocations
 
-__all__ = [
-    "choose",
-    "compress",
-    "diag_indices",
-    "diag_indices_from",
-    "diagonal",
-    "extract",
-    "fill_diagonal",
-    "flatnonzero",
-    "indices",
-    "iterable",
-    "ix_",
-    "mask_indices",
-    "ndindex",
-    "nonzero",
-    "place",
-    "put",
-    "put_along_axis",
-    "putmask",
-    "ravel_multi_index",
-    "select",
-    "take",
-    "take_along_axis",
-    "tril_indices",
-    "tril_indices_from",
-    "triu_indices",
-    "triu_indices_from",
-    "unravel_index",
-]
-
 
 def _ravel_multi_index_checks(multi_index, dims, order):
     dpnp.check_supported_arrays_type(*multi_index)

dpnp/dpnp_iface_linearalgebra.py

@@ -54,21 +54,6 @@
     dpnp_vecdot,
 )
 
-__all__ = [
-    "dot",
-    "einsum",
-    "einsum_path",
-    "inner",
-    "kron",
-    "matmul",
-    "matvec",
-    "outer",
-    "tensordot",
-    "vdot",
-    "vecdot",
-    "vecmat",
-]
-
 
 # TODO: implement a specific scalar-array kernel
 def _call_multiply(a, b, out=None, outer_calc=False):

dpnp/dpnp_iface_logic.py

@@ -55,36 +55,6 @@
 
 from .dpnp_utils import get_usm_allocations
 
-__all__ = [
-    "all",
-    "allclose",
-    "any",
-    "array_equal",
-    "array_equiv",
-    "equal",
-    "greater",
-    "greater_equal",
-    "isclose",
-    "iscomplex",
-    "iscomplexobj",
-    "isfinite",
-    "isfortran",
-    "isinf",
-    "isnan",
-    "isneginf",
-    "isposinf",
-    "isreal",
-    "isrealobj",
-    "isscalar",
-    "less",
-    "less_equal",
-    "logical_and",
-    "logical_not",
-    "logical_or",
-    "logical_xor",
-    "not_equal",
-]
-
 
 def _isclose_scalar_tol(a, b, rtol, atol, equal_nan):
     """

dpnp/dpnp_iface_manipulation.py

@@ -40,8 +40,6 @@
 """
 
 
-# pylint: disable=duplicate-code
-
 import math
 import operator
 import warnings
@@ -97,67 +95,6 @@ class UniqueInverseResult(NamedTuple):
     inverse_indices: dpnp.ndarray
 
 
-__all__ = [
-    "append",
-    "array_split",
-    "asarray_chkfinite",
-    "asfarray",
-    "atleast_1d",
-    "atleast_2d",
-    "atleast_3d",
-    "broadcast_arrays",
-    "broadcast_shapes",
-    "broadcast_to",
-    "can_cast",
-    "column_stack",
-    "concat",
-    "concatenate",
-    "copyto",
-    "delete",
-    "dsplit",
-    "dstack",
-    "expand_dims",
-    "flip",
-    "fliplr",
-    "flipud",
-    "hsplit",
-    "hstack",
-    "insert",
-    "matrix_transpose",
-    "moveaxis",
-    "ndim",
-    "pad",
-    "permute_dims",
-    "ravel",
-    "repeat",
-    "require",
-    "reshape",
-    "resize",
-    "result_type",
-    "roll",
-    "rollaxis",
-    "rot90",
-    "row_stack",
-    "shape",
-    "size",
-    "split",
-    "squeeze",
-    "stack",
-    "swapaxes",
-    "tile",
-    "transpose",
-    "trim_zeros",
-    "unique",
-    "unique_all",
-    "unique_counts",
-    "unique_inverse",
-    "unique_values",
-    "unstack",
-    "vsplit",
-    "vstack",
-]
-
-
 def _check_stack_arrays(arrays):
     """Validate a sequence type of arrays to stack."""