Added __repr__ to ProbDist

reachtarunhere · reachtarunhere · commit d81be44c4aa3 · 2016-06-30T22:41:37.000+05:30
diff --git a/probability.py b/probability.py
@@ -79,6 +79,9 @@ def show_approx(self, numfmt='%.3g'):
         return ', '.join([('%s: ' + numfmt) % (v, p)
                           for (v, p) in sorted(self.prob.items())])
 
+    def __repr__(self):
+        return "P(%s)" % self.varname
+
 
 class JointProbDist(ProbDist):
     """A discrete probability distribute over a set of variables.
