Skip to content

Commit ac252fd

Browse files
committed
Using issubdtype in choose integral indices check
1 parent 65f7320 commit ac252fd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpnp/dpnp_iface_indexing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def choose(a, choices, out=None, mode="wrap"):
239239

240240
inds = dpnp.get_usm_ndarray(a)
241241
ind_dt = inds.dtype
242-
if ind_dt.kind not in "ui":
242+
if not dpnp.issubdtype(ind_dt, dpnp.integer):
243243
raise ValueError("input index array must be of integer data type")
244244

245245
choices, queues, usm_types = _build_choices_list(choices)

0 commit comments

Comments
 (0)