Fix typo in __setitem__ error message

pymc-devs · Oct 31, 2024 · 03e5f77 · 03e5f77
1 parent 4f7d709
commit 03e5f77
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pytensor/tensor/variable.py b/pytensor/tensor/variable.py
@@ -598,7 +598,7 @@ def is_empty_array(val):
 
     def __setitem__(self, key, value):
         raise TypeError(
-            "TensorVariable does not support item assignment. Use the output of `set` or `add` instead."
+            "TensorVariable does not support item assignment. Use the output of `x[idx].set` or `x[idx].inc` instead."
         )
 
     def take(self, indices, axis=None, mode="raise"):

diff --git a/tests/tensor/test_variable.py b/tests/tensor/test_variable.py
@@ -1,3 +1,4 @@
+import re
 from copy import copy
 
 import numpy as np
@@ -444,7 +445,7 @@ def test_set_inc(self):
     def test_set_item_error(self):
         x = matrix("x")
 
-        msg = "Use the output of `set` or `add` instead."
+        msg = re.escape("Use the output of `x[idx].set` or `x[idx].inc` instead.")
         with pytest.raises(TypeError, match=msg):
             x[0] = 5
         with pytest.raises(TypeError, match=msg):