Global Significant Figures Formatting

Author: saeedashrraf

ivy/__init__.py

@@ -37,6 +37,10 @@ class Dtype:
     pass
 
 
+array_significant_figures_stack = list()
+array_decimal_values_stack = list()
+
+
 # global constants
 _MIN_DENOMINATOR = 1e-12
 _MIN_BASE = 1e-5
@@ -359,3 +363,125 @@ class StaticContainer(ContainerBase):
 
 if "IVY_BACKEND" in os.environ:
     ivy.set_framework(os.environ["IVY_BACKEND"])
+
+# Array Significant Figures #
+
+
+def _assert_array_significant_figures_formatting(sig_figs):
+    assert isinstance(sig_figs, int)
+    assert sig_figs > 0
+
+
+def _sf(x, sig_fig=3):
+    if isinstance(x, np.bool_):
+        return x
+    f = float(np.format_float_positional(
+        x, precision=sig_fig, unique=False, fractional=False, trim='k'
+    ))
+    if np.issubdtype(type(x), np.uint) :
+        f = np.uint(f)
+    if np.issubdtype(type(x), np.int) :
+        f = np.int(f)
+    x = f
+    return x
+
+
+vec_sig_fig = np.vectorize(_sf)
+vec_sig_fig.__name__ = 'vec_sig_fig'
+
+
+def array_significant_figures(sig_figs=None):
+    """Summary.
+
+    Parameters
+    ----------
+    sig_figs
+        optional int, number of significant figures to be shown when printing
+
+    Returns
+    -------
+    ret
+
+    """
+    if ivy.exists(sig_figs):
+        _assert_array_significant_figures_formatting(sig_figs)
+        return sig_figs
+    global array_significant_figures_stack
+    if not array_significant_figures_stack:
+        ret = 3 
+    else:
+        ret = array_significant_figures_stack[-1]
+    return ret
+
+
+def set_array_significant_figures(sig_figs):
+    """Summary.
+
+    Parameters
+    ----------
+    sig_figs
+        optional int, number of significant figures to be shown when printing
+
+    """
+    _assert_array_significant_figures_formatting(sig_figs)
+    global array_significant_figures_stack
+    array_significant_figures_stack.append(sig_figs)
+
+
+def unset_array_significant_figures():
+    """"""
+    global array_significant_figures_stack
+    if array_significant_figures_stack:
+        array_significant_figures_stack.pop(-1)
+
+# Decimal Values #
+
+
+def _assert_array_decimal_values_formatting(dec_vals):
+    assert isinstance(dec_vals, int)
+    assert dec_vals >= 0
+
+
+def array_decimal_values(dec_vals=None):
+    """Summary.
+
+    Parameters
+    ----------
+    dec_vals
+        optional int, number of decimal values to be shown when printing
+
+    Returns
+    -------
+    ret
+
+    """
+    if ivy.exists(dec_vals):
+        _assert_array_decimal_values_formatting(dec_vals)
+        return dec_vals
+    global array_decimal_values_stack
+    if not array_decimal_values_stack:
+        ret = None
+    else:
+        ret = array_decimal_values_stack[-1]
+    return ret
+
+
+def set_array_decimal_values(dec_vals):
+    """Summary.
+
+    Parameters
+    ----------
+    dec_vals
+        optional int, number of significant figures to be shown when printing
+
+    """
+    _assert_array_decimal_values_formatting(dec_vals)
+    global array_decimal_values_stack
+    array_decimal_values_stack.append(dec_vals)
+
+
+def unset_array_decimal_values():
+    """"""
+    global array_decimal_values_stack
+    if array_decimal_values_stack:
+        array_decimal_values_stack.pop(-1)

ivy/array/__init__.py

@@ -5,6 +5,7 @@
 from operator import mul
 
 # local
+import numpy as np
 from . import conversions
 from .conversions import *
 
@@ -198,9 +199,13 @@ def __array_wrap__(self, *args, **kwargs):
 
     @_native_wrapper
     def __repr__(self):
-        return (
+        sig_fig = ivy.array_significant_figures()
+        dec_vals = ivy.array_decimal_values()
+        rep = ivy.vec_sig_fig(ivy.to_numpy(self._data),sig_fig) if self._size > 0 else ivy.to_numpy(self._data)
+        with np.printoptions(precision=dec_vals):
+            return (
             self._pre_repr
-            + ivy.to_numpy(self._data)
+            + rep
             .__repr__()[:-1]
             .partition(", dtype")[0]
             .partition(", dev")[0]

ivy/func_wrapper.py

@@ -76,6 +76,7 @@
     "fn_array_spec",
     "insert_into_nest_at_index",
     "insert_into_nest_at_indices",
+    "vec_sig_fig"
 ]
 METHODS_W_CONT_SUPPORT = [
     "multi_head_attention",