diff options
| author | Stephan Hoyer <shoyer@google.com> | 2018-11-13 09:38:07 -0800 |
|---|---|---|
| committer | Stephan Hoyer <shoyer@google.com> | 2018-11-13 20:24:31 -0800 |
| commit | 4d24bbda32d133d51940b0691bd9b428d4198eaa (patch) | |
| tree | c018ac1ada700494f1bb404a8ae9f1346720ebfe /numpy/lib/arraysetops.py | |
| parent | cd39348e8593dc2b41e2516fbdd8a69b0f0bda6e (diff) | |
| download | numpy-4d24bbda32d133d51940b0691bd9b428d4198eaa.tar.gz | |
ENH: set correct __module__ for objects in numpy's public API
Fixes GH-12271
Tests verify that everything in ``dir(numpy)`` either has ``__module__`` set to
``'numpy'``, or appears in an explicit whitelist of undocumented functions and
exported bulitins. These should eventually be documented or removed.
I also identified a handful of functions for which I had accidentally not setup
dispatch for with ``__array_function__`` before, because they were listed under
"ndarray methods" in ``_add_newdocs.py``. I guess that should be a lesson in
trusting code comments :).
Diffstat (limited to 'numpy/lib/arraysetops.py')
| -rw-r--r-- | numpy/lib/arraysetops.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/numpy/lib/arraysetops.py b/numpy/lib/arraysetops.py index 850e20123..fd64ecbd6 100644 --- a/numpy/lib/arraysetops.py +++ b/numpy/lib/arraysetops.py @@ -479,6 +479,11 @@ def setxor1d(ar1, ar2, assume_unique=False): return aux[flag[1:] & flag[:-1]] +def _in1d_dispatcher(ar1, ar2, assume_unique=None, invert=None): + return (ar1, ar2) + + +@array_function_dispatch(_in1d_dispatcher) def in1d(ar1, ar2, assume_unique=False, invert=False): """ Test whether each element of a 1-D array is also present in a second array. |