Skip to content

Commit e501dc7

Browse files
committed
revert ndarray
1 parent 8808c39 commit e501dc7

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

sharpy/ndarray.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -52,21 +52,17 @@ def to_device(self, device=""):
5252
return ndarray(self._t.to_device(device))
5353

5454
def __getitem__(self, key):
55-
# print('__getitem__', key)
5655
key = key if isinstance(key, tuple) else (key,)
5756
key = [slicefy(x) for x in key]
5857
return ndarray(self._t.__getitem__(key))
5958

6059
def __setitem__(self, key, value):
61-
# print('__setitem__', key)
62-
# key = key if isinstance(key, tuple) else (key,)
63-
# key = [slicefy(x) for x in key]
64-
# print('key', key)
65-
array = self.__getitem__(key)
60+
key = key if isinstance(key, tuple) else (key,)
61+
key = [slicefy(x) for x in key]
6662
if isinstance(value, ndarray) and value._t.dtype != self._t.dtype:
6763
raise ValueError(
6864
f"Mismatching data type in setitem: {value._t.dtype}, expecting {self._t.dtype}"
6965
)
70-
array._t.__setitem__(
66+
self._t.__setitem__(
7167
key, value._t if isinstance(value, ndarray) else value
7268
)

0 commit comments

Comments
 (0)