Simplify string expression for Constant Variables

Author: ricardoV94

pytensor/graph/basic.py

@@ -763,13 +763,20 @@ def signature(self):
         return (self.type, self.data)
 
     def __str__(self):
-        if self.name is not None:
-            return self.name
-        else:
-            name = str(self.data)
-            if len(name) > 20:
-                name = name[:10] + "..." + name[-10:]
-            return f"{type(self).__name__}{{{name}}}"
+        data_str = str(self.data)
+        if len(data_str) > 20:
+            data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
+
+        if self.name is None:
+            return data_str
+
+        return f"{self.name}{{{data_str}}}"
+
+    def __repr__(self):
+        data_str = repr(self.data)
+        if len(data_str) > 20:
+            data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
+        return f"{type(self).__name__}({repr(self.type)}, data={data_str})"
 
     def clone(self, **kwargs):
         return self

pytensor/tensor/var.py

@@ -1023,21 +1023,6 @@ def __init__(self, type: _TensorTypeType, data, name=None):
 
         Constant.__init__(self, new_type, data, name)
 
-    def __str__(self):
-        unique_val = get_unique_value(self)
-        if unique_val is not None:
-            val = f"{self.data.shape} of {unique_val}"
-        else:
-            val = f"{self.data}"
-        if len(val) > 20:
-            val = val[:10].strip() + " ... " + val[-10:].strip()
-
-        if self.name is not None:
-            name = self.name
-        else:
-            name = "TensorConstant"
-        return f"{name}{{{val}}}"
-
     def signature(self):
         return TensorConstantSignature((self.type, self.data))
 

tests/graph/rewriting/test_basic.py

@@ -166,7 +166,7 @@ def test_constant(self):
         e = op1(op1(x, y), y)
         g = FunctionGraph([y], [e])
         OpKeyPatternNodeRewriter((op1, z, "1"), (op2, "1", z)).rewrite(g)
-        assert str(g) == "FunctionGraph(Op1(Op2(y, z), y))"
+        assert str(g) == "FunctionGraph(Op1(Op2(y, z{2}), y))"
 
     def test_constraints(self):
         x, y, z = MyVariable("x"), MyVariable("y"), MyVariable("z")