O}m= d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z e Z
dZd Z e e d e e
d
ej d d Zd
Z ddZ ddZdS )z;Implementation of __array_function__ overrides from NEP-18. N )
set_module)
getargspec)
add_docstring_get_implementing_args_ArrayFunctionDispatcherar like : array_like, optional
Reference object to allow the creation of arrays which are not
NumPy arrays. If an array-like passed in as ``like`` supports
the ``__array_function__`` protocol, the result will be defined
by it. In this case, it ensures the creation of an array object
compatible with that passed in via this argument.c ^ | j %| j dt | _ | S )Nz${ARRAY_FUNCTION_LIKE})__doc__replacearray_function_like_doc)
public_apis h/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/overrides.pyset_array_function_like_docr s4 % ' / 7 7 $ #
a
Class to wrap functions with checks for __array_function__ overrides.
All arguments are required, and can only be passed by position.
Parameters
----------
dispatcher : function or None
The dispatcher function that returns a single sequence-like object
of all arguments relevant. It must have the same signature (except
the default values) as the actual implementation.
If ``None``, this is a ``like=`` dispatcher and the
``_ArrayFunctionDispatcher`` must be called with ``like`` as the
first (additional and positional) argument.
implementation : function
Function that implements the operation on NumPy arrays without
overrides. Arguments passed calling the ``_ArrayFunctionDispatcher``
will be forwarded to this (and the ``dispatcher``) as if using
``*args, **kwargs``.
Attributes
----------
_implementation : function
The original implementation passed in.
ar
Collect arguments on which to call __array_function__.
Parameters
----------
relevant_args : iterable of array-like
Iterable of possibly array-like arguments to check for
__array_function__ methods.
Returns
-------
Sequence of arguments with __array_function__ methods, in the order in
which they should be called.
ArgSpecz args varargs keywords defaultsc t t | }t t | }|j |j k s{|j |j k sk|j |j k s[t |j t |j k s1|j z* 5 56O O$888 '?+B B B (O,D D D
%.
/
/
/*
+
+, ,
)
5
$-
.
.
)
*
*+ + ;=K L M M M #/ #w_5M1N1N'N N N 9 : :
: 0/ N Nr TFc fd}|S )ax Decorator for adding dispatch with the __array_function__ protocol.
See NEP-18 for example usage.
Parameters
----------
dispatcher : callable or None
Function that when called like ``dispatcher(*args, **kwargs)`` with
arguments from the NumPy function call returns an iterable of
array-like arguments to check for ``__array_function__``.
If `None`, the first argument is used as the single `like=` argument
and not passed on. A function implementing `like=` must call its
dispatcher with `like` as the first non-keyword argument.
module : str, optional
__module__ attribute to set on new function, e.g., ``module='numpy'``.
By default, module is copied from the decorated function.
verify : bool, optional
If True, verify the that the signature of the dispatcher and decorated
function signatures match exactly: all required and optional arguments
should appear in order with the same names, but the default values for
all optional arguments should be ``None``. Only disable verification
if the dispatcher's signature needs to deviate for some particular
reason, e.g., because the function has a signature like
``func(*args, **kwargs)``.
docs_from_dispatcher : bool, optional
If True, copy docs from the dispatcher function onto the dispatched
function, rather than from the implementation. This is useful for
functions defined in C, which otherwise don't have docstrings.
Returns
-------
Function suitable for decorating the implementation of a NumPy function.
c r]t | nJ| j }|j |j z dz
}|j | }|dk s |j dk rt d| d rt
| j t | } t j
| | }|_ t
| |S )N liker zX__array_function__ expects `like=` to be the last argument and a keyword-only argument. z does not seem to comply.)r __code__ co_argcountco_kwonlyargcount co_varnamesr r r
r functoolswraps
__module__ARRAY_FUNCTIONSadd)r colast_argr
r docs_from_dispatchermoduleverifys r decoratorz*array_function_dispatch..decorator s F%*>:FFFF $, >B,@@1D >(3v%%)=)B)B&E)E E EF F F
> .**< = = =-j.II
4Y_^44Z@@
$*J !J'''r )r r/ r0 r. r1 s ```` r array_function_dispatchr3 i s7 J 8 r c fd}|S )zDLike array_function_dispatcher, but with function arguments flipped.c < t | S )N)r0 r. )r3 )r r. r r/ r0 s r r1 z1array_function_from_dispatcher..decorator s9 7& v!57 7 77EG G Gr r2 )r r/ r0 r. r1 s ```` r array_function_from_dispatcherr6 s> G G G G G G G G r )NNTF)NTT)r
collectionsr' os_utilsr _utils._inspectr numpy.core._multiarray_umathr r r setr* r r
namedtupler r r3 r6 r2 r r r> s\ A A ( ( ( ( ( (F F F F F F F F F F #%%=
<
$ !+
,L
M
M: : :, BF16A A A AJ HL r