diff --git a/test/unit/test_indexing.py b/test/unit/test_indexing.py index 9fad1cd381..1b38e52190 100644 --- a/test/unit/test_indexing.py +++ b/test/unit/test_indexing.py @@ -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 ''' diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 64cc898975..5dfa3e9891 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -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`.