Skip to content

Commit 2f6554d

Browse files
committed
Give MultiTrace a more informative repr output
Add number of chains, number of iterations, and number of variables to repr output.
1 parent 1f2a639 commit 2f6554d

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

pymc/backends/base.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,11 @@ def __init__(self, traces):
143143
raise ValueError("Chains are not unique.")
144144
self._traces[trace.chain] = trace
145145

146+
def __repr__(self):
147+
template = '<{}: {} chains, {} iterations, {} variables>'
148+
return template.format(self.__class__.__name__,
149+
self.nchains, len(self), len(self.varnames))
150+
146151
@property
147152
def nchains(self):
148153
return len(self._traces)

0 commit comments

Comments
 (0)