File tree Expand file tree Collapse file tree 1 file changed +3
-7
lines changed
Expand file tree Collapse file tree 1 file changed +3
-7
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments