U
|
O ±d}! ã @ s¦ d Z ddlZddlZddlZddlmZmZmZ ddlm Z e
|
e ej dd¡ZdZdd Zeed
|
eed e d d ¡Zdd Zdd ZdddZdddZdS )z;Implementation of __array_function__ overrides from NEP-18.é N)Ú add_docstringÚimplement_array_functionÚ_get_implementing_args)Ú
|
getargspecZ!NUMPY_EXPERIMENTAL_ARRAY_FUNCTIONé ar 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 C s | j d k r| j dt¡| _ | S )Nz${ARRAY_FUNCTION_LIKE})Ú__doc__ÚreplaceÚarray_function_like_doc)Ú
|
public_api© r úKd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/overrides.pyÚset_array_function_like_doc s
|
þr aû
|
Implement a function with checks for __array_function__ overrides.
|
|
All arguments are required, and can only be passed by position.
|
|
Parameters
|
----------
|
implementation : function
|
Function that implements the operation on NumPy array without
|
overrides when called like ``implementation(*args, **kwargs)``.
|
public_api : function
|
Function exposed by NumPy's public API originally called like
|
``public_api(*args, **kwargs)`` on which arguments are now being
|
checked.
|
relevant_args : iterable
|
Iterable of arguments to check for __array_function__ methods.
|
args : tuple
|
Arbitrary positional arguments originally passed into ``public_api``.
|
kwargs : dict
|
Arbitrary keyword arguments originally passed into ``public_api``.
|
|
Returns
|
-------
|
Result from calling ``implementation()`` or an ``__array_function__``
|
method, as appropriate.
|
|
Raises
|
------
|
TypeError : if no implementation is found.
|
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 C s¤ t t| }t t| }|j|jksn|j|jksn|j|jksnt|jt|jksn|jdk rzt|jt|jkrztd| |jdk r |jdt|j kr tddS )z:Verify that a dispatcher function has the right signature.NzGimplementation and dispatcher for %s have different function signatures©NzBdispatcher functions can only use None for default argument values) r r ÚargsÚvarargsÚkeywordsÚboolÚdefaultsÚlenÚ RuntimeError)ÚimplementationÚ
|
dispatcherZimplementation_specZdispatcher_specr r r Úverify_matching_signaturesX s.
|
ÿ
|
þÿýûÿúÿ
|
r c s fdd}|S )zÉDecorator for overriding __module__ on a function or class.
|
|
Example usage::
|
|
@set_module('numpy')
|
def example():
|
pass
|
|
assert example.__module__ == 'numpy'
|
c s d k r | _ | S r )Ú
|
__module__)Úfunc©Úmoduler r Ú decoratory s z set_module.<locals>.decoratorr )r r r r r Ú
|
set_modulen s r TFc s0 t s fdd}|S fdd}|S )a Decorator for adding dispatch with the __array_function__ protocol.
|
|
See NEP-18 for example usage.
|
|
Parameters
|
----------
|
dispatcher : callable
|
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__``.
|
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 s" rt | j d k r | _| S r )r r r ©r )r Údocs_from_dispatcherr r r r ¢ s
|
z*array_function_dispatch.<locals>.decoratorc sf rt r t j t ¡ fddjj jdd_d k r\_ _ S )Nc
|
s z | |}W n` t k
|
rn } zB|jd } j}t|trD| |¡sF | |j¡}t |d W 5 d }~X Y nX t|| |S )Nr )Ú TypeErrorr Ú__name__Ú
|
isinstanceÚstrÚ
|
startswithr r )r ÚkwargsZ relevant_argsÚexcÚmsgZ disp_nameZnew_msg)r r r
|
Úuse_liker r r
|
± s"
|
|
þz>array_function_dispatch.<locals>.decorator.<locals>.public_apiz <__array_function__ internals>)Úco_nameÚ co_filename)
|
r r r Ú functoolsÚwrapsÚ__code__r r# r Z_implementationr ©r r! r r* Úverify)r r
|
r r ª s
|
þ)ÚARRAY_FUNCTION_ENABLED)r r r1 r! r* r r r0 r Úarray_function_dispatch s
|
!,r3 c s fdd}|S )zDLike array_function_dispatcher, but with function arguments flipped.c s t | dS )N)r1 r! )r3 )r ©r! r r r1 r r r Ý s þþz1array_function_from_dispatcher.<locals>.decoratorr )r r r1 r! r r r4 r Ú array_function_from_dispatcherÙ s r5 )NTFF)NTT)r Ú collectionsr- ÚosZ numpy.core._multiarray_umathr r r Znumpy.compat._inspectr r ÚintÚenvironÚgetr2 r r Ú
|
namedtupler r r r3 r5 r r r r Ú<module> s< ÿÿ þ#þ ÿ
|
Z ÿ
|
