diff options
Diffstat (limited to 'numpy/core/overrides.py')
-rw-r--r-- | numpy/core/overrides.py | 9 |
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, """ |