restrict assignment to base (#3809)

* restrict assignment to base

* add some restrictions there

* more restrictions
This commit is contained in:
George Hotz
2024-03-18 15:33:06 -07:00
committed by GitHub
parent 20681d5c4a
commit 4c4d3cb3e3
6 changed files with 17 additions and 11 deletions

View File

@@ -178,7 +178,7 @@ class TestAssign(unittest.TestCase):
b.realize()
ba1 = a.lazydata.base.realized
bb1 = b.lazydata.base.realized
with self.assertRaises(RuntimeError):
with self.assertRaises((RuntimeError, AssertionError)):
a = a.permute(1,0)
a += b
a.realize()