-jedZddlZddlmZddlmZddlmZddlm Z ddl m Z e dZ d e d eed e fged e fffd Z d edeeefd eefdZddgZd eefdZdS)z Internal helpers N)Callable)wraps) signature) ModuleType)TypeVar_Txpreturn.cddtdtfdtdtfffd }|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. f.r c4tdtdtdtffd }t}|d|jD}|jdjd|_||_|S) Nargskwargsr c|di|S)Nr )rrr r s ZC:\Projects\DSCons\.venv\Lib\site-packages\sklearn\externals\array_api_compat\_internal.py wrapped_fz(get_xp..inner..wrapped_f!s1d,r,V,, ,c$g|] \}}|dk |S)r r).0ipars r z)get_xp..inner..'s!NNN3AIIIIIr) parametersz$Array API compatibility wrapper for zk. See the corresponding documentation in NumPy/CuPy and/or the array API specification for more details. ) robjectrreplaceritems__doc____name__ __signature__)r rsignew_sigr s` rinnerzget_xp..inner s q -V -v -& - - - - - -  -ll++NN#.*>*>*@*@NNN     $!%&Z!!!I #* r)rr)r r#s` rget_xpr$sH"#r'"(37*;* Lrmod_nameglobals_c<tj|}i}td|jd|t |D]:}|ds#t ||rt||||<;||t|S)zLImport everything from module, updating globals(). Returns __all__. zfrom z import *_) importlib import_moduleexecrdir startswithhasattrgetattrupdatelist)r%r&modobjsns r clone_moduler58s  !( + +C D ( ( ( ($/// XX&&||C   &WS!__ &c1ooDG OOD ::rr$r5ctS)N)__all__rrr__dir__r8Ls Nr)rr)collections.abcr functoolsrinspectrtypesrtypingrrr$strdictrr1r5r7r8rrrr@s1$$$$$$ WT]]&z&hb(9':HS"W