Skip to content

Commit 9c09ffc

Browse files
committed
address review comments
1 parent 5164f63 commit 9c09ffc

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

dpnp/dpnp_iface_logic.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@
5353
import dpnp.backend.extensions.ufunc._ufunc_impl as ufi
5454
from dpnp.dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc
5555

56+
from .dpnp_array import dpnp_array
5657
from .dpnp_utils import get_usm_allocations
5758

5859

@@ -1271,7 +1272,7 @@ def isin(
12711272
)
12721273
usm_element = dpnp.get_usm_ndarray(element)
12731274
usm_test = dpnp.get_usm_ndarray(test_elements)
1274-
return dpnp.get_result_array(
1275+
return dpnp_array._create_from_usm_ndarray(
12751276
dpt.isin(
12761277
usm_element,
12771278
usm_test,

dpnp/tests/test_logic.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
1+
import dpctl
12
import numpy
23
import pytest
4+
from dpctl.utils import ExecutionPlacementError
35
from numpy.testing import (
46
assert_allclose,
57
assert_array_equal,
@@ -880,12 +882,18 @@ def test_isin_empty(self):
880882
assert_equal(result, expected)
881883

882884
def test_isin_errors(self):
883-
a = dpnp.arange(5)
884-
b = dpnp.arange(3)
885+
q1 = dpctl.SyclQueue()
886+
q2 = dpctl.SyclQueue()
887+
888+
a = dpnp.arange(5, sycl_queue=q1)
889+
b = dpnp.arange(3, sycl_queue=q2)
885890

886891
# unsupported type for elements or test_elements
887892
with pytest.raises(TypeError):
888-
dpnp.isin(dict(), b)
893+
dpnp.isin(dict(), a)
889894

890895
with pytest.raises(TypeError):
891896
dpnp.isin(a, dict())
897+
898+
with pytest.raises(ExecutionPlacementError):
899+
dpnp.isin(a, b)

0 commit comments

Comments
 (0)