Merge pull request numpy#27255 from ngoldbaum/fix-printoptions

BUG: revert unintended change in the return value of set_printoptions

Author: seberg

numpy/_core/arrayprint.py

@@ -280,6 +280,15 @@ def set_printoptions(precision=None, threshold=None, edgeitems=None,
     array([ 0.  ,  1.11,  2.22, ...,  7.78,  8.89, 10.  ])
 
     """
+    _set_printoptions(precision, threshold, edgeitems, linewidth, suppress,
+                      nanstr, infstr, formatter, sign, floatmode,
+                      legacy=legacy, override_repr=override_repr)
+
+
+def _set_printoptions(precision=None, threshold=None, edgeitems=None,
+                      linewidth=None, suppress=None, nanstr=None,
+                      infstr=None, formatter=None, sign=None, floatmode=None,
+                      *, legacy=None, override_repr=None):
     new_opt = _make_options_dict(precision, threshold, edgeitems, linewidth,
                                  suppress, nanstr, infstr, sign, formatter,
                                  floatmode, legacy)
@@ -293,8 +302,7 @@ def set_printoptions(precision=None, threshold=None, edgeitems=None,
     if updated_opt['legacy'] == 113:
         updated_opt['sign'] = '-'
 
-    token = format_options.set(updated_opt)
-    return token
+    return format_options.set(updated_opt)
 
 
 @set_module('numpy')
@@ -378,8 +386,9 @@ def printoptions(*args, **kwargs):
     --------
     set_printoptions, get_printoptions
 
-    """ 
-    token = set_printoptions(*args, **kwargs)
+    """
+    token = _set_printoptions(*args, **kwargs)
+
     try:
         yield get_printoptions()
     finally:

numpy/_core/tests/test_arrayprint.py

@@ -627,8 +627,9 @@ def teardown_method(self):
     def test_basic(self):
         x = np.array([1.5, 0, 1.234567890])
         assert_equal(repr(x), "array([1.5       , 0.        , 1.23456789])")
-        np.set_printoptions(precision=4)
+        ret = np.set_printoptions(precision=4)
         assert_equal(repr(x), "array([1.5   , 0.    , 1.2346])")
+        assert ret is None
 
     def test_precision_zero(self):
         np.set_printoptions(precision=0)