We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 295de83 commit 365a5f2Copy full SHA for 365a5f2
dpnp/dpnp_iface_indexing.py
@@ -130,13 +130,7 @@ def _build_choices_list(choices):
130
if dpnp.is_supported_array_type(choices):
131
queues = [choices.sycl_queue]
132
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]
+ choices = [dpnp.get_usm_ndarray(chc) for chc in dpnp.unstack(choices)]
140
elif isinstance(choices, (tuple, list)):
141
queues = []
142
usm_types = []
0 commit comments