Skip to content

Commit 80ac085

Browse files
committed
Skip key normalization in _resolve_indexing_state
1 parent 76c5c85 commit 80ac085

1 file changed

Lines changed: 0 additions & 12 deletions

File tree

heat/core/dndarray.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -332,18 +332,6 @@ def _resolve_indexing_state(
332332
backwards_transpose_axes=tuple(range(arr.ndim)),
333333
)
334334

335-
# normalize index components
336-
if isinstance(key, DNDarray):
337-
if key.dtype not in (ht_bool, ht_uint8) and key.split is None:
338-
key = key.larray.to(torch.int64)
339-
elif isinstance(key, (list, tuple)):
340-
key = type(key)(
341-
k.larray.to(torch.int64)
342-
if isinstance(k, DNDarray) and k.dtype not in (ht_bool, ht_uint8) and k.split is None
343-
else k
344-
for k in key
345-
)
346-
347335
# 1D boolean mask resolution
348336
first = key[0] if isinstance(key, tuple) and len(key) >= 1 else key
349337
if isinstance(first, (DNDarray, torch.Tensor, np.ndarray)) and arr.ndim >= 1:

0 commit comments

Comments
 (0)