diff options
author | Stephan Hoyer <shoyer@google.com> | 2018-10-23 07:53:58 -0700 |
---|---|---|
committer | Stephan Hoyer <shoyer@google.com> | 2018-10-23 07:57:40 -0700 |
commit | 8bab96faf2cb740536712e49e92e133626087018 (patch) | |
tree | b16d0f4d121d588ca3219b95d3c4304059cddacd /numpy/lib/index_tricks.py | |
parent | 42a5bddc937ef02175d52a3ab3b3da6229e1ecb2 (diff) | |
download | numpy-8bab96faf2cb740536712e49e92e133626087018.tar.gz |
MAINT: set preferred __module__ for numpy functions
Diffstat (limited to 'numpy/lib/index_tricks.py')
-rw-r--r-- | numpy/lib/index_tricks.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/numpy/lib/index_tricks.py b/numpy/lib/index_tricks.py index 26243d231..0cdf78c6d 100644 --- a/numpy/lib/index_tricks.py +++ b/numpy/lib/index_tricks.py @@ -1,5 +1,6 @@ from __future__ import division, absolute_import, print_function +import functools import sys import math @@ -13,10 +14,14 @@ from . import function_base import numpy.matrixlib as matrixlib from .function_base import diff from numpy.core.multiarray import ravel_multi_index, unravel_index -from numpy.core.overrides import array_function_dispatch +from numpy.core import overrides from numpy.lib.stride_tricks import as_strided +array_function_dispatch = functools.partial( + overrides.array_function_dispatch, module='numpy') + + __all__ = [ 'ravel_multi_index', 'unravel_index', 'mgrid', 'ogrid', 'r_', 'c_', 's_', 'index_exp', 'ix_', 'ndenumerate', 'ndindex', 'fill_diagonal', |