summaryrefslogtreecommitdiff
path: root/numpy/lib/index_tricks.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/index_tricks.py
parent42a5bddc937ef02175d52a3ab3b3da6229e1ecb2 (diff)
downloadnumpy-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.py7
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',