
    = j1                        S SK r S SKJrJrJrJrJr  S SKrSSK	7  SSK
J	r	  SSK
Jr  \	R                  rSrS r/ S	QrS
 r\" SS/SSS9r\" SSSS9\" SSSS9\" SSSS9\" SSSS9\" SSSS9\" SS9\" SSS/SSS9S.r\	R                   HR  r\" \	\5      r\" \\S-   5      r\R/                  \\5      r\" \(       a  \" \\5      " \5      O\5      r\\" 5       \'   MT     g)    N)is_cupyis_jaxscipy_namespace_forSCIPY_ARRAY_APIxp_capabilities   )*)_ndimage_api)_delegatorsndimagec                    ^ [        U [        R                  [        R                  -  5      (       a  TR	                  U 5      $ [        U [
        [        -  5      (       a  [        U 5      " U4S jU  5       5      $ U $ )z4Convert arrays/scalars hiding in the sequence `arg`.c              3   <   >#    U  H  n[        UT5      v   M     g 7fN)_maybe_convert_arg).0xxps     |/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/scipy/ndimage/_support_alternative_backends.py	<genexpr>%_maybe_convert_arg.<locals>.<genexpr>   s     @Cq+Ar22Cs   )
isinstancenpndarraygenericasarraylisttupletype)argr   s    `r   r   r      sV    #rzzBJJ.//zz#	C	&	&Cy@C@@@
    )distance_transform_bfdistance_transform_cdtfind_objectsgeometric_transformvectorized_filterc                    ^ ^ U U4S jnU$ )Nc                 L   >^  [         R                  " T 5      UU U4S j5       nU$ )Nc            	        >^ T" U 0 UD6m[        T5      (       aJ  TR                  [        ;  a6  SS KnUR	                  ST 35      n[        UTR                  5      nU" U 0 UD6$ [        T5      (       aE  TR                  S:X  a5  [        T5      n[        UT5      n[        UTR                  5      nU" U 0 UD6$ T" U 0 UD6n[        U[        R                  [        R                  -  5      (       a  TR                  U5      $ [        U[        5      (       a  U$ [        U[        5      (       aG  UR                  5        V	V
s0 s H)  u  pU	R!                  5       [#        U4S jU
 5       5      _M+     sn
n	$ Uc  U$ [%        UT5      $ s  sn
n	f )Nr   zcupyx.scipy.map_coordinatesc              3   F   >#    U  H  nTR                  U5      v   M     g 7fr   )r   )r   vvr   s     r   r   >delegate_xp.<locals>.inner.<locals>.wrapper.<locals>.<genexpr>E   s     'C2

2s   !)r   __name__CUPY_BLOCKLIST	importlibimport_modulegetattrr   r   r   r   r   r   r   intdictitemsitemr   r   )argskwdsr/   cupyx_module
cupyx_funcspx
jax_modulejax_funcresultkvr   	delegatorfuncmodule_names              @r   wrapper+delegate_xp.<locals>.inner.<locals>.wrapper'   sf   D)D)B r{{t}}NB (66k]7ST$\4==A
!404001B B)"-$S+6
":t}}=... t,t,fbjj2::&=>>::f--,,!M--
 $*<<>#1CA %'C'C"CC#1  ^!M .fb99s   0F)	functoolswraps)rA   rC   r@   rB   s   ` r   innerdelegate_xp.<locals>.inner&   s&    		%	: 
%	:L r     )r@   rB   rG   s   `` r   delegate_xprJ   %   s    (R Lr    TcupyF)cpu_only
exceptionsallow_dask_computejax_jit)rL   rN   rO   )out_of_scopez	jax.numpy)r$   r#   r!   r"   r%   generate_binary_structurer)   
_signature)rE   scipy._lib._array_apir   r   r   r   r   numpyr   r
    r   __all__MODULE_NAMEr   r.   rJ   default_capabilitiescapabilities_dict	func_namer1   	bare_funcr@   getcapabilitiesfvarsrI   r    r   <module>r`      sK        


 *X 'vh4 
 +$ $$ -$ .$ )$ "1d!C&6;"7# 0 %%Ii0IY%=>I$((4HIL 	I{+I6 )	A
 DF9 &r    