Improve string representation of Assert Ops

Author: ricardoV94

pytensor/raise_op.py

@@ -48,7 +48,13 @@ def __init__(self, exc_type, msg=""):
         self.msg = msg
 
     def __str__(self):
-        return f"CheckAndRaise{{{self.exc_type}({self.msg})}}"
+        name = self.__class__.__name__
+        exc_name = self.exc_type.__name__
+        if len(self.msg) > 30:
+            msg = self.msg[:27] + "..."
+        else:
+            msg = self.msg
+        return f"{name}{{raises={exc_name}, msg='{msg}'}}"
 
     def __eq__(self, other):
         if type(self) is not type(other):
@@ -195,7 +201,11 @@ def __init__(self, msg="PyTensor Assert failed!"):
         super().__init__(AssertionError, msg)
 
     def __str__(self):
-        return f"Assert{{msg={self.msg}}}"
+        if len(self.msg) > 30:
+            msg = self.msg[:27] + "..."
+        else:
+            msg = self.msg
+        return f"Assert{{msg='{msg}'}}"
 
 
 assert_op = Assert()