Output sparse ternary differently.

malb · Oct 29, 2024 · 9ef4365 · 9ef4365
1 parent cbee849
commit 9ef4365
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/estimator/nd.py b/estimator/nd.py
@@ -510,6 +510,23 @@ def support_size(self, fraction=1.0):
         n, p, m = len(self), self.p, self.m
         return ceil(binomial(n, p) * binomial(n - p, m) * RR(fraction))
 
+    def __str__(self):
+        """
+        EXAMPLE::
+
+            >>> from estimator import *
+            >>> ND.SparseTernary(20, 20, n=100)
+            T(p=20, m=20, n=100)
+
+        """
+        if self.n:
+            return f"T(p={self.p}, m={self.m}, n={int(self.n)})"
+        else:
+            return f"T(p={int(self.p)}, m={int(self.m)})"
+
+    def __repr__(self):
+        return str(self)
+
 
 def SparseBinary(hw, n=None):
     """