o i+ @sddlZddlmZmZmZmZddlZddlTddl mZddl m Z ej Z dZ dd Z ej D]ZeeeZee ed ZerGe ee eneZeee<q0dS) N)is_cupyis_jaxscipy_namespace_forSCIPY_ARRAY_API)*) _ndimage_api) _delegatorsZndimagecsfdd}|S)Ncstfdd}|S)Nc s|i|tr$ddl}|d}t|j}||i|StrCjdkrCt}t|}t|j}||i|S|i|}t|tj tj frX |St|t r_|St|t rofdd|DS|duru|St|fdd|DS)Nrz cupyx.scipy.Zmap_coordinatescs,i|]\}}|tfdd|DqS)c3s|]}|VqdSN)asarray).0ZvvZxpe/home/kim/smarthome/.venv/lib/python3.10/site-packages/scipy/ndimage/_support_alternative_backends.py /szIdelegate_xp..inner..wrapper...)itemtuple)r kvr rr .sz?delegate_xp..inner..wrapper..c3s*|]}t|tjr|n|VqdSr ) isinstancenpndarrayr )r xr rrr7s  z>delegate_xp..inner..wrapper..)r importlib import_modulegetattr__name__rrrrrZgenericr intdictitemstype) argskwdsrZ cupyx_moduleZ cupyx_funcZspxZ jax_moduleZjax_funcresult) delegatorfunc module_namer rwrappers2       z+delegate_xp..inner..wrapper) functoolswraps)r&r(r%r')r&rinners(zdelegate_xp..innerr)r%r'r,rr+r delegate_xps+r- _signature)r)Zscipy._lib._array_apirrrrnumpyrrr __all__Z MODULE_NAMEr- func_namerZ bare_funcr%fvarsrrrrs$   /