Q3jdZddlZddlmZddlmZddlmZddlm Z ddl m Z e dZ d e d eed e fged e fffd Z d edeeefd eefdZddgZd eefdZy)z Internal helpers N)Callable)wraps) signature) ModuleType)TypeVar_Txpreturn.cPdtdtfdtdtfffd }|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 cPtdtdtdtffd }t}|j|jj Dcgc] \}}|dk7s |c}}}|j djd|_||_|Scc}}w) Nargskwargsr c|di|S)Nr )rrr r s Y/DATA/.local/lib/python3.12/site-packages/sklearn/externals/array_api_compat/_internal.py wrapped_fz(get_xp..inner..wrapped_f!sd,r,V, ,r ) 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 rsigiparnew_sigr s` rinnerzget_xp..inner s q -V -v -& -  -l++*-..*>*>*@N*@3AI*@N     $%%%&ZZL1!I #* Os  B" +B" )rr)r r s` rget_xpr!s/"#r'"(37*;* Lrmod_nameglobals_ctj|}i}td|jd|t |D]0}|j drt ||s"t||||<2|j|t|S)zLImport everything from module, updating globals(). Returns __all__. zfrom z import *_) importlib import_moduleexecrdir startswithhasattrgetattrupdatelist)r"r#modobjsns r clone_moduler28s{  ! !( +C D5i ($/ X||C WS!_c1oDG OOD :rr!r2ctS)N)__all__rrr__dir__r5Ls Nr)rr&collections.abcr functoolsrinspectrtypesrtypingrrr!strdictrr.r2r4r5rrrr=s$ T]&z&hb(9':HS"W