o  h+ @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) _delegatorsndimagecsfdd}|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.map_coordinatescs,i|]\}}|tfdd|DqS)c3s|]}|VqdSN)asarray).0vvxp_/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/ndimage/_support_alternative_backends.py /szIdelegate_xp..inner..wrapper...)itemtuple)rkvrrr .sz?delegate_xp..inner..wrapper..c3s*|]}t|tjr|n|VqdSr ) isinstancenpndarrayr )rxrrrr7s  z>delegate_xp..inner..wrapper..)r importlib import_modulegetattr__name__rrrrrgenericr intdictitemstype) argskwdsr cupyx_module cupyx_funcspx jax_modulejax_funcresult) delegatorfunc module_namerrwrappers2       z+delegate_xp..inner..wrapper) functoolswraps)r0r2r/r1)r0rinners(zdelegate_xp..innerr)r/r1r6rr5r delegate_xps+r7 _signature)r3scipy._lib._array_apirrrrnumpyrrr __all__ MODULE_NAMEr7 func_namer bare_funcr/fvarsrrrrs$   /