summaryrefslogtreecommitdiff
path: root/numpy/core/numeric.py
diff options
context:
space:
mode:
authorSebastian Berg <sebastian@sipsolutions.net>2022-05-27 15:10:59 -0700
committerSebastian Berg <sebastian@sipsolutions.net>2022-06-15 11:42:02 -0700
commitc855cecce28d5f925c5b2e015a0bdfd1aa472590 (patch)
tree35b7e271aee7b3d5df416ecea5e3bd55f4d1ecbb /numpy/core/numeric.py
parent09d407a3cd24b712f8a40748262e00188e8b8efa (diff)
downloadnumpy-c855cecce28d5f925c5b2e015a0bdfd1aa472590.tar.gz
API: Expose `get_promotion_state` and `set_promotion_state`
We need to be able to query the state for testing, probably should be renamed before the end, but need to have something for now.
Diffstat (limited to 'numpy/core/numeric.py')
-rw-r--r--numpy/core/numeric.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/numpy/core/numeric.py b/numpy/core/numeric.py
index 38d85da6e..f81cfebce 100644
--- a/numpy/core/numeric.py
+++ b/numpy/core/numeric.py
@@ -17,7 +17,7 @@ from .multiarray import (
fromstring, inner, lexsort, matmul, may_share_memory,
min_scalar_type, ndarray, nditer, nested_iters, promote_types,
putmask, result_type, set_numeric_ops, shares_memory, vdot, where,
- zeros, normalize_axis_index)
+ zeros, normalize_axis_index, get_promotion_state, set_promotion_state)
from . import overrides
from . import umath
@@ -54,7 +54,8 @@ __all__ = [
'False_', 'True_', 'bitwise_not', 'CLIP', 'RAISE', 'WRAP', 'MAXDIMS',
'BUFSIZE', 'ALLOW_THREADS', 'ComplexWarning', 'full', 'full_like',
'matmul', 'shares_memory', 'may_share_memory', 'MAY_SHARE_BOUNDS',
- 'MAY_SHARE_EXACT', 'TooHardError', 'AxisError']
+ 'MAY_SHARE_EXACT', 'TooHardError', 'AxisError',
+ 'get_promotion_state', 'set_promotion_state']
@set_module('numpy')