diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/core/einsumfunc.py | 4 | ||||
| -rw-r--r-- | numpy/core/shape_base.py | 6 |
2 files changed, 7 insertions, 3 deletions
diff --git a/numpy/core/einsumfunc.py b/numpy/core/einsumfunc.py index 3ffb152e1..d9f88cb1c 100644 --- a/numpy/core/einsumfunc.py +++ b/numpy/core/einsumfunc.py @@ -700,7 +700,7 @@ def _einsum_path_dispatcher(*operands, **kwargs): return operands -@array_function_dispatch(_einsum_path_dispatcher) +@array_function_dispatch(_einsum_path_dispatcher, module='numpy') def einsum_path(*operands, **kwargs): """ einsum_path(subscripts, *operands, optimize='greedy') @@ -1001,7 +1001,7 @@ def _einsum_dispatcher(*operands, **kwargs): # Rewrite einsum to handle different cases -@array_function_dispatch(_einsum_dispatcher) +@array_function_dispatch(_einsum_dispatcher, module='numpy') def einsum(*operands, **kwargs): """ einsum(subscripts, *operands, out=None, dtype=None, order='K', diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py index c9f8ebccb..71a23f438 100644 --- a/numpy/core/shape_base.py +++ b/numpy/core/shape_base.py @@ -7,9 +7,13 @@ import functools import operator from . import numeric as _nx +from . import overrides from .numeric import array, asanyarray, newaxis from .multiarray import normalize_axis_index -from .overrides import array_function_dispatch + + +array_function_dispatch = functools.partial( + overrides.array_function_dispatch, module='numpy') def _atleast_1d_dispatcher(*arys): |