Skip to content

Commit 98193d1

Browse files
committed
Simplify string expression for Constant Variables
1 parent e3e0424 commit 98193d1

File tree

4 files changed

+95
-103
lines changed

4 files changed

+95
-103
lines changed

pytensor/graph/basic.py

+14-7
Original file line numberDiff line numberDiff line change
@@ -763,13 +763,20 @@ def signature(self):
763763
return (self.type, self.data)
764764

765765
def __str__(self):
766-
if self.name is not None:
767-
return self.name
768-
else:
769-
name = str(self.data)
770-
if len(name) > 20:
771-
name = name[:10] + "..." + name[-10:]
772-
return f"{type(self).__name__}{{{name}}}"
766+
data_str = str(self.data)
767+
if len(data_str) > 20:
768+
data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
769+
770+
if self.name is None:
771+
return data_str
772+
773+
return f"{self.name}{{{data_str}}}"
774+
775+
def __repr__(self):
776+
data_str = repr(self.data)
777+
if len(data_str) > 20:
778+
data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
779+
return f"{type(self).__name__}({repr(self.type)}, data={data_str})"
773780

774781
def clone(self, **kwargs):
775782
return self

pytensor/tensor/var.py

-15
Original file line numberDiff line numberDiff line change
@@ -1023,21 +1023,6 @@ def __init__(self, type: _TensorTypeType, data, name=None):
10231023

10241024
Constant.__init__(self, new_type, data, name)
10251025

1026-
def __str__(self):
1027-
unique_val = get_unique_value(self)
1028-
if unique_val is not None:
1029-
val = f"{self.data.shape} of {unique_val}"
1030-
else:
1031-
val = f"{self.data}"
1032-
if len(val) > 20:
1033-
val = val[:10].strip() + " ... " + val[-10:].strip()
1034-
1035-
if self.name is not None:
1036-
name = self.name
1037-
else:
1038-
name = "TensorConstant"
1039-
return f"{name}{{{val}}}"
1040-
10411026
def signature(self):
10421027
return TensorConstantSignature((self.type, self.data))
10431028

0 commit comments

Comments
 (0)