address review comments

Author: ndgrigorian

dpnp/dpnp_iface_logic.py

@@ -1200,24 +1200,27 @@ def isfortran(a):
 def isin(element, test_elements, assume_unique=False, invert=False):
     """
     Calculates ``element in test_elements``, broadcasting over `element` only.
-    Returns a boolean array of the same shape as `element` that is True
-    where an element of `element` is in `test_elements` and False otherwise.
+    Returns a boolean array of the same shape as `element` that is ``True``
+    where an element of `element` is in `test_elements` and ``False``
+    otherwise.
+
+    For full documentation refer to :obj:`numpy.isin`.
 
     Parameters
     ----------
-    element : {array_like, dpnp.ndarray, usm_ndarray}
+    element : {dpnp.ndarray, usm_ndarray, scalar}
         Input array.
-    test_elements : {array_like, dpnp.ndarray, usm_ndarray}
+    test_elements : {dpnp.ndarray, usm_ndarray, scalar}
         The values against which to test each value of `element`.
-        This argument is flattened if it is an array or array_like.
-        See notes for behavior with non-array-like parameters.
+        This argument is flattened if it is an array.
     assume_unique : bool, optional
         Ignored
     invert : bool, optional
-        If True, the values in the returned array are inverted, as if
-        calculating `element not in test_elements`. Default is False.
+        If ``True``, the values in the returned array are inverted, as if
+        calculating `element not in test_elements`.
         ``dpnp.isin(a, b, invert=True)`` is equivalent to (but faster
         than) ``dpnp.invert(dpnp.isin(a, b))``.
+        Default: ``False``.
 
 
     Returns
@@ -1259,28 +1262,18 @@ def isin(element, test_elements, assume_unique=False, invert=False):
     """
 
     dpnp.check_supported_arrays_type(element, test_elements, scalar_type=True)
-    if dpnp.isscalar(element):
-        usm_element = dpt.asarray(
-            element,
-            sycl_queue=test_elements.sycl_queue,
-            usm_type=test_elements.usm_type,
-        )
-        usm_test = dpnp.get_usm_ndarray(test_elements)
-    elif dpnp.isscalar(test_elements):
-        usm_test = dpt.asarray(
-            test_elements,
-            sycl_queue=element.sycl_queue,
-            usm_type=element.usm_type,
-        )
-        usm_element = dpnp.get_usm_ndarray(element)
-    else:
-        usm_element = dpnp.get_usm_ndarray(element)
-        usm_test = dpnp.get_usm_ndarray(test_elements)
+    usm_element = dpnp.as_usm_ndarray(
+        element, usm_type=element.usm_type, sycl_queue=element.sycl_queue
+    )
+    usm_test = dpnp.as_usm_ndarray(
+        test_elements,
+        usm_type=test_elements.usm_type,
+        sycl_queue=test_elements.sycl_queue,
+    )
     return dpnp.get_result_array(
         dpt.isin(
             usm_element,
             usm_test,
-            assume_unique=assume_unique,
             invert=invert,
         )
     )