8000 ENH: implement ``__skip_array_function__`` attribute for NEP-18 by shoyer · Pull Request #13389 · numpy/numpy · GitHub
[go: up one dir, main page]
Skip to content

ENH: implement __skip_array_function__ attribute for NEP-18 #13389

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
May 18, 2019
Merged

ENH: implement __skip_array_function__ attribute for NEP-18 #13389

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
afda748
ENH: implement __numpy_implementation__ attribute for NEP-18
shoyer Apr 10, 2019
389c19d
MNT: fix failure on Python 3.5
shoyer Apr 23, 2019
804f71b
Rename to __skip_array_function__
shoyer May 11, 2019
07452c8
Merge branch 'master' into implement-numpy-implementation
shoyer May 13, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Rename to __skip_array_function__
  • Loading branch information
@shoyer
shoyer committed May 11, 2019
commit 804f71bb42fef775a85a52366dc4bd1a22c130a8
2 changes: 1 addition & 1 deletion numpy/core/overrides.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ def public_api(*args, **kwargs):
if module is not None:
public_api.__module__ = module

public_api.__numpy_implementation__ = implementation
public_api.__skip_array_function__ = implementation

return public_api
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This will not merge cleanly after #13529

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fixed


Expand Down
2 changes: 1 addition & 1 deletion numpy/core/src/multiarray/arrayfunction_override.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,7 @@ array_function_method_impl(PyObject *func, PyObject *types, PyObject *args,
}
}

implementation = PyObject_GetAttr(func, npy_ma_str_numpy_implementation);
implementation = PyObject_GetAttr(func, npy_ma_str_skip_array_function);
if (implementation == NULL) {
return NULL;
}
Expand Down
8 changes: 4 additions & 4 deletions numpy/core/src/multiarray/multiarraymodule.c
View file
Open in desktop
8000
Original file line number Diff line number Diff line change
Expand Up @@ -4493,7 +4493,7 @@ NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_array_prepare = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_array_wrap = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_array_finalize = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_ufunc = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_numpy_implementation = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_skip_array_function = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_order = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_copy = NULL;
NPY_VISIBILITY_HIDDEN PyObject * npy_ma_str_dtype = NULL;
Expand All @@ -4509,8 +4509,8 @@ intern_strings(void)
npy_ma_str_array_wrap = PyUString_InternFromString("__array_wrap__");
npy_ma_str_array_finalize = PyUString_InternFromString("__array_finalize__");
npy_ma_str_ufunc = PyUString_InternFromString("__array_ufunc__");
npy_ma_str_numpy_implementation = PyUString_InternFromString(
"__numpy_implementation__");
npy_ma_str_skip_array_function = PyUString_InternFromString(
"__skip_array_function__");
npy_ma_str_order = PyUString_InternFromString("order");
npy_ma_str_copy = PyUString_InternFromString("copy");
npy_ma_str_dtype = PyUString_InternFromString("dtype");
Expand All @@ -4520,7 +4520,7 @@ intern_strings(void)

return npy_ma_str_array && npy_ma_str_array_prepare &&
npy_ma_str_array_wrap && npy_ma_str_array_finalize &&
npy_ma_str_ufunc && npy_ma_str_numpy_implementation &&
npy_ma_str_ufunc && npy_ma_str_skip_array_function &&
npy_ma_str_order && npy_ma_str_copy && npy_ma_str_dtype &&
npy_ma_str_ndmin && npy_ma_str_axis1 && npy_ma_str_axis2;
}
Expand Down
2 changes: 1 addition & 1 deletion numpy/core/src/multiarray/multiarraymodule.h
10000
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_array_prepare;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_array_wrap;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_array_finalize;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_ufunc;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_numpy_implementation;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_skip_array_function;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_order;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_copy;
NPY_VISIBILITY_HIDDEN extern PyObject * npy_ma_str_dtype;
Expand Down
14 changes: 7 additions & 7 deletions numpy/core/tests/test_overrides.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -190,17 +190,17 @@ class OverrideSub(np.ndarray):
result = np.concatenate((array, override_sub))
assert_equal(result, expected.view(OverrideSub))

def test_numpy_implementation(self):
assert_(dispatched_one_arg.__numpy_implementation__
def test_skip_array_function(self):
assert_(dispatched_one_arg.__skip_array_function__
is dispatched_one_arg.__wrapped__)

def test_no_wrapper(self):
# This shouldn't happen unless a user intentionally calls
# __array_function__ with invalid arguments, but check that we raise
# an appropriate error all the same.
array = np.array(1)
func = dispatched_one_arg.__numpy_implementation__
with assert_raises_regex(AttributeError, '__numpy_implementation__'):
func = dispatched_one_arg.__skip_array_function__
with assert_raises_regex(AttributeError, '__skip_array_function__'):
array.__array_function__(func=func, types=(np.ndarray,),
args=(array,), kwargs={})

Expand Down Expand Up @@ -386,7 +386,7 @@ def _(array):
assert_equal(np.sum(MyArray()), 'yes')

def test_sum_implementation_on_list(self):
assert_equal(np.sum.__numpy_implementation__([1, 2, 3]), 6)
assert_equal(np.sum.__skip_array_function__([1, 2, 3]), 6)

def test_sum_on_mock_array(self):

Expand All @@ -410,7 +410,7 @@ def __array__(self, *args, **kwargs):

proxy = ArrayProxy(mock.Mock(spec=ArrayProxy))
proxy.value.__array__.return_value = np.array(2)
result = np.sum.__numpy_implementation__(proxy)
result = np.sum.__skip_array_function__(proxy)
assert_equal(result, 2)
# TODO: switch to proxy.value.__array__.assert_called() and
# proxy.value.__array_function__.assert_not_called() once we drop
Expand All @@ -426,7 +426,7 @@ def sum(self, axis, out):
return 'summed'

def __array_function__(self, func, types, args, kwargs):
return func.__numpy_implementation__(*args, **kwargs)
return func.__skip_array_function__(*args, **kwargs)

# note: the internal implementation of np.sum() calls the .sum() method
assert_equal(np.sum(MyArray()), 'summed')
0