From edf929ec9d3292aad382536655a3f526ed264331 Mon Sep 17 00:00:00 2001 From: ayanhan <64448894+ayanhan@users.noreply.github.com> Date: Thu, 4 Dec 2025 17:53:08 +0900 Subject: [PATCH] fix: add __delitem__ to Tensor with proper TypeError (#13561) --- test/unit/test_indexing.py | 5 ++--- tinygrad/tensor.py | 3 +++ 2 files changed, 5 insertions(+), 3 deletions(-) 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`.