Skip to content

Commit 797c86b

Browse files
committed
Always unstack choices array regardless of dimension
1 parent eff36de commit 797c86b

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

dpnp/dpnp_iface_indexing.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -129,13 +129,7 @@ def _build_choices_list(choices):
129129
if dpnp.is_supported_array_type(choices):
130130
queues = [choices.sycl_queue]
131131
usm_types = [choices.usm_type]
132-
choices_sh = choices.shape
133-
if len(choices_sh) > 1:
134-
choices = [
135-
dpnp.get_usm_ndarray(chc) for chc in dpnp.unstack(choices)
136-
]
137-
else:
138-
choices = [choices]
132+
choices = [dpnp.get_usm_ndarray(chc) for chc in dpnp.unstack(choices)]
139133
elif isinstance(choices, (tuple, list)):
140134
queues = []
141135
usm_types = []

0 commit comments

Comments
 (0)