summaryrefslogtreecommitdiff
path: root/numpy/lib/twodim_base.py
diff options
context:
space:
mode:
authorStephan Hoyer <shoyer@google.com>2018-10-23 07:53:58 -0700
committerStephan Hoyer <shoyer@google.com>2018-10-23 07:57:40 -0700
commit8bab96faf2cb740536712e49e92e133626087018 (patch)
treeb16d0f4d121d588ca3219b95d3c4304059cddacd /numpy/lib/twodim_base.py
parent42a5bddc937ef02175d52a3ab3b3da6229e1ecb2 (diff)
downloadnumpy-8bab96faf2cb740536712e49e92e133626087018.tar.gz
MAINT: set preferred __module__ for numpy functions
Diffstat (limited to 'numpy/lib/twodim_base.py')
-rw-r--r--numpy/lib/twodim_base.py8
1 files changed, 7 insertions, 1 deletions
diff --git a/numpy/lib/twodim_base.py b/numpy/lib/twodim_base.py
index aff6cddde..a05e68375 100644
--- a/numpy/lib/twodim_base.py
+++ b/numpy/lib/twodim_base.py
@@ -3,12 +3,14 @@
"""
from __future__ import division, absolute_import, print_function
+import functools
+
from numpy.core.numeric import (
absolute, asanyarray, arange, zeros, greater_equal, multiply, ones,
asarray, where, int8, int16, int32, int64, empty, promote_types, diagonal,
nonzero
)
-from numpy.core.overrides import array_function_dispatch
+from numpy.core import overrides
from numpy.core import iinfo, transpose
@@ -18,6 +20,10 @@ __all__ = [
'tril_indices_from', 'triu_indices', 'triu_indices_from', ]
+array_function_dispatch = functools.partial(
+ overrides.array_function_dispatch, module='numpy')
+
+
i1 = iinfo(int8)
i2 = iinfo(int16)
i4 = iinfo(int32)