diff --git a/pytensor/raise_op.py b/pytensor/raise_op.py index 1da5cc24dd..dd20c7acf0 100644 --- a/pytensor/raise_op.py +++ b/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) != 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()