github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 22:48:25 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: add __delitem__ to Tensor with proper TypeError (#13561)

Browse Source
This commit is contained in:
ayanhan
2025-12-04 17:53:08 +09:00
committed by GitHub
parent 9411ecedc4
commit edf929ec9d
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit/test_indexing.py
View File

@@ -176,9 +176,8 @@ class TestIndexing(unittest.TestCase):
self.assertRaises(IndexError, lambda: reference[0.0, ..., 0.0:2.0])
self.assertRaises(IndexError, lambda: reference[0.0, :, 0.0])
# TODO: delitem
# def delitem(): del reference[0]
# self.assertRaises(TypeError, delitem)
def delitem(): del reference[0]
self.assertRaises(TypeError, delitem)
# TODO setitem backward
'''

3
tinygrad/tensor.py
View File

@@ -1257,6 +1257,9 @@ class Tensor(OpMixin):
v = v.cast(res.dtype)._broadcast_to(_broadcast_shape(res.shape, v.shape)).contiguous()
res.assign(v).realize()
def __delitem__(self, indices) -> None:
raise TypeError("Tensor does not support deleting items")
def gather(self:Tensor, dim:int, index:Tensor) -> Tensor:
"""
Gathers values along an axis specified by `dim`.