o Yi@s(dZddlmZddlmZddZdS)z Internal helpers )wraps) signaturecsfdd}|S)a2 Decorator to automatically replace xp with the corresponding array module. Use like import numpy as np @get_xp(np) def func(x, /, xp, kwarg=None): return xp.func(x, kwarg=kwarg) Note that xp must be a keyword argument and come after all non-keyword arguments. cs^tfdd}tjfddjDd}|jdur*djd|_||_|S)Ncs|di|S)Nxp)argskwargs)frrL/tmp/pip-target-1s0edx8b/lib/python/scipy/_lib/array_api_compat/_internal.py wrapped_fsz(get_xp..inner..wrapped_fcsg|] }|dkrj|qSr parameters).0i)sigrr sz)get_xp..inner..r z$Array API compatibility wrapper for zk. See the corresponding documentation in NumPy/CuPy and/or the array API specification for more details. )rrreplacer __doc____name__ __signature__)rr new_sigr )rrr inners  zget_xp..innerr)rrrr r get_xps rN)r functoolsrinspectrrrrrr s