Skip to content

Commit 365a5f2

Browse files
committed
Always unstack choices array regardless of dimension
1 parent 295de83 commit 365a5f2

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
@@ -130,13 +130,7 @@ def _build_choices_list(choices):
130130
if dpnp.is_supported_array_type(choices):
131131
queues = [choices.sycl_queue]
132132
usm_types = [choices.usm_type]
133-
choices_sh = choices.shape
134-
if len(choices_sh) > 1:
135-
choices = [
136-
dpnp.get_usm_ndarray(chc) for chc in dpnp.unstack(choices)
137-
]
138-
else:
139-
choices = [choices]
133+
choices = [dpnp.get_usm_ndarray(chc) for chc in dpnp.unstack(choices)]
140134
elif isinstance(choices, (tuple, list)):
141135
queues = []
142136
usm_types = []

0 commit comments

Comments
 (0)