Move is_type_supported() to dpnp_iface_random.py and remove from public API

Author: vlad-perevezentsev

dpnp/__init__.py

@@ -262,7 +262,6 @@
     iinfo,
     isdtype,
     issubdtype,
-    is_type_supported,
 )
 from .dpnp_iface_manipulation import can_cast, result_type
 from .dpnp_iface_types import dtype
@@ -760,7 +759,6 @@
     "iinfo",
     "isdtype",
     "issubdtype",
-    "is_type_supported",
     "result_type",
 ]
 

dpnp/dpnp_iface_types.py

@@ -311,11 +311,3 @@ def issubdtype(arg1, arg2):
     """
 
     return numpy.issubdtype(arg1, arg2)
-
-
-def is_type_supported(obj_type):
-    """Return True if type is supported by DPNP python level."""
-
-    if obj_type in (float64, float32, int64, int32):
-        return True
-    return False

dpnp/random/dpnp_iface_random.py

@@ -69,6 +69,14 @@ def _get_random_state(device=None, sycl_queue=None):
     return _dpnp_random_states[sycl_queue]
 
 
+def _is_type_supported(obj_type):
+    """Return True if type is supported by dpnp.random"""
+
+    if obj_type in (dpnp.float64, dpnp.float32, dpnp.int64, dpnp.int32):
+        return True
+    return False
+
+
 def beta(a, b, size=None):
     """
     Draw samples from a Beta distribution.
@@ -1587,7 +1595,7 @@ def shuffle(x1):
                 "Running on CUDA is currently not supported"
             )
 
-        if not dpnp.is_type_supported(x1_desc.dtype):
+        if not _is_type_supported(x1_desc.dtype):
             pass
         else:
             dpnp_rng_shuffle(x1_desc).get_pyobj()