summaryrefslogtreecommitdiff
path: root/numpy/core/overrides.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/core/overrides.py')
-rw-r--r--numpy/core/overrides.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/numpy/core/overrides.py b/numpy/core/overrides.py
index c2b5fb7fa..31d1c1d74 100644
--- a/numpy/core/overrides.py
+++ b/numpy/core/overrides.py
@@ -34,6 +34,15 @@ def set_array_function_like_doc(public_api):
return public_api
+def array_function_dispatch_like(func, *args, **kwargs):
+ if not hasattr(kwargs['like'], '__array_function__'):
+ raise ValueError(
+ 'The `like` object must implement the `__array_function__` '
+ 'protocol.'
+ )
+ return func(*args, **kwargs)
+
+
add_docstring(
implement_array_function,
"""