
    biV                    n   U d dl mZ d dlZd dlZdaded<   ddZd Zg dZ	 d dl	m
Z d d	l	mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d d
lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z% d dl&m'Z' d dl(m)Z)m*Z* d dl+m,Z, d dl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@ d dlAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQ d dlRmSZSmTZTmUZU d dlVmWZWmXZX d dlYmZZZ d dl[m\Z\m]Z] d dl^m_Z_m`Z` d dlambZb d dlcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9 d dl:m;Z;m<Z<m=Z=m>Z> d dl?m@Z@  e       r!d dlAmc mBZC  ej                  eC      ZC e       rd  ZH	 d d!lAm.Z. d d"lAmIZ$ d d#lAmZmZmZmZmSZSmBZBmZmZmZmZmZmZmZmZmTZTm0Z0m1Z1mZmZmZmZm*Z*mZmZmZm5Z5mZmZmZmZmZmZmZmZmJZJm;Z;mEZEmZmZmZmZmZmZmZmZmZmZmZmZmZm6Z6mZm<Z<mHZHmZmZmUZUm Z mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmJZJmZmZmZmZmZmZmZmZmZmZm:Z:m\Z\mZmZmZmZmZmZmZm Z m!Z!m=Z=mLZLm"Z"m#Z#m$Z$m%Z%m%Z%m&Z&mbZbm'Z'mZm(Z(mNZNm)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m;Z;m3Z3m4Z4m5Z5m6Z6m7Z7m>Z>mQZQ d d$lcmdZdmeZemkZkmlZlmnZnmoZomtZtmuZumwZwmxZxmzZzm{Z{m|Z|m}Z}mZmZmZmZ  eHd%      Z eHd&      Z  eHd'      Z! eHd(      Z" eHd)      Z# eHd*      Z) eHd+      Z, eHd,      Z2 eHd-      Z3 eHd.      Z4 eHd/      Z7 eHd0      Z8 eHd1      Z9 eHd2      Z< eHd3      Z= eHd4      Z> eHd5      Z? eHd6      Z@ eHd7      ZC eHd8      ZD eHd9      ZF eHd:      ZG eHd;      ZI eHd<      ZK eHd=      ZM eHd>      ZO eHd?      ZP eHd@      ZW eHdA      ZX eHdB      ZZ eHdC      Z` eHdD      Zf eHdE      Zg eHdF      Zh eHdG      Zi eHdH      Zj eHdI      Zm eHdJ      Zp eHdK      Zq eHdL      Zr eHdM      Zs eHdN      Zv eHdO      Z eHdP      Z eHdQ      Z eHdR      Z eHdS      Z eHdT      Z eHdU      Z eHdV      Z eHdW      Z eHdX      Z eHdY      Z eHdZ      Z eHd[      Z eHd\      Z eHd]      Z eHd^      Z eHd_      Z eHd`      Z eHda      Z eHdb      Z eHdc      Z eHdd      Z eHde      Z eHdf      Z eHdg      Z eHdh      Z eHdi      Z eHdj      Z eHdk      Z eHdl      Z eHdm      Z eHdn      Z eHdo      Z eHdp      Z eHdq      Z eHdr      Z eHds      Z eHdt      Z eHdu      Z eHdv      Z eHdw      Z eHdx      Z eHdy      Z eHdz      Z eHd{      Z eHd|      Z eHd}      Z eHd~      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Z eHd      Zd dl8m9Z9 d dl?m@Z@ yy# eE$ r ZdZF eE eGe      dz   eFz         edZ[ww xY w# eE$ r d dlmZC  ej                  eC      ZCY yw xY w)    )annotationsNzbool | NoneARRAY_EXPR_ENABLEDc                     dd l } | j                  j                  d      }t        B|du rt        du s|du r,t        du r$t	        j
                  d| dt        rdnd d	       t        S t        ||      S d      S )
Nr   zarray.query-planningTFz0The 'array.query-planning' config is now set to z , but query planning is already enableddisabledzV. The query-planning config can only be changed before `dask.array` is first imported!)daskconfiggetr   warningswarnbool)r   use_array_exprs     N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/__init__.py_array_expr_enabledr   
   s     [[__%;<N%d"'9U'Be#(:d(BMMB!""B 29
C D22 "!."<HH%HH    c                     t               S )N)r    r   r   array_expr_enabledr   !   s      r   (#  r   	complex64
complex128eeuler_gammafloat32float64infint8int16int32int64nannewaxispiuint8uint16uint32uint64backendsfftliblinalgmaoverlaprandomr   shuffleatop	blockwiseregister_chunk_typeArrayPerformanceWarning
asanyarrayasarrayblockbroadcast_arraysbroadcast_toconcatenate
from_arrayfrom_delayedfrom_npy_stack	from_zarr
map_blocksstackstoreto_hdf5to_npy_stackto_zarrunify_chunksarangediagdiagonal
empty_likeeyefromfunction	full_likeindiceslinspacemeshgrid	ones_likepadrepeattiletri
zeros_likeapply_gufunc	as_gufuncgufuncmoveaxisrollaxisoptimizemap_overlappushnanpercentile
percentilerechunkallanyargmaxargminargtopkcumprodcumsummaxmeanmedianminmoment	nanargmax	nanargmin
nancumprod	nancumsumnanmaxnanmean	nanmediannanminnanprodnanquantilenanstdnansumnanvarprodquantile	reductionstdsumtopktracevarreshapereshape_blockwiseallcloseappendapply_along_axisapply_over_axesargwherearoundarray
atleast_1d
atleast_2d
atleast_3daveragebincountchoosecoarsencompresscorrcoefcount_nonzerocovdeletediffdigitizedotdstackediff1deinsumexpand_dimsextractflatnonzeroflipfliplrflipudgradient	histogramhistogram2dhistogramddhstackinsertiscloseisinisnullmatmulndimnonzeronotnullouter	piecewiseptpravelravel_multi_indexresult_typerollrot90roundsearchsortedselectshapesqueezeswapaxestake	tensordot	transposetriltril_indicestril_indices_fromtriutriu_indicestriu_indices_fromunion1duniqueunravel_indexvdotvstackwherefrom_tiledb	to_tiledbabsabsoluteaddanglearccosarccosharcsinarcsinharctanarctan2arctanhbitwise_andbitwise_not
bitwise_orbitwise_xorcbrtceilclipconjcopysigncoscoshdeg2raddegreesdividedivmodequalexpexp2expm1fabsfixfloat_powerfloorfloor_dividefmaxfminfmodfrexp
frompyfuncgreatergreater_equalhypoti0imaginvert	iscomplexisfiniteisinfisnanisneginfisposinfisrealldexp
left_shiftless
less_equalloglog1plog2log10	logaddexp
logaddexp2logical_andlogical_not
logical_orlogical_xormaximumminimummodmodfmultiply
nan_to_numnegative	nextafter	not_equalpositivepowerrad2degradiansreal
reciprocal	remainderright_shiftrintsignsignbitsinsincsinhspacingsqrtsquaresubtracttantanhtrue_dividetrunc	assert_eqemptyfulloneszeroscompute)bool_)r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   )r'   r(   r)   r*   r+   r,   r-   )r.   )r/   r0   )r1   )r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   )rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   )rU   rV   rW   )rX   rY   )rZ   )r[   r\   )r]   r^   )r_   )!r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   )r   r   )Ir   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )r   r   )br   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r	  r
  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r+  r,  r-  r.  r/  )r0  )r1  r2  r3  r4  )r5  zDask array requirements are not installed.

Please either conda or pip install as follows:

  conda install dask                 # either conda install
  python -m pip install "dask[array]" --upgrade  # or python -m pip installz

c                      fd}|S )Nc                 "    t        d d      )Nz	Function z" is not implemented for dask-expr.)NotImplementedError)argskwargs	attr_names     r   
inner_funcz/raise_not_implemented_error.<locals>.inner_func~  s    %I;&HI r   r   )r<  r=  s   ` r   raise_not_implemented_errorr>  }  s    	
 r   )r2   )_overlap)}r   r   r   r   rU   rE   r   r   r   r   r   r   r   r   rV   r4   r5   r   r   r   r   r0   r   r   r   r9   r   r   r   r   r   r   r   r   elemwiser1  rH   r   r   r   r   r   r   r   r   r   r   r   r   r   r:   r   r2  rK   r   r   rW   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  rM   r  r  r	  r
  r  r  r  r  r  r  r>   r[   r  r  r  r  r  r  r  r  r  r3  rO   r  r  r  r  r-   r  r_   r  r{   r   rQ   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r?   r+  r,  r-  r.  r/  r4  rT   )r`   ra   rg   rh   rj   rk   rp   rq   rs   rt   rv   rw   rx   ry   r{   r|   r}   r   r'   r(   r)   r*   r+   r/   r1   r6   r7   r8   r;   r<   r=   r@   rA   rB   rC   rD   rF   rG   rI   rJ   rL   rN   rP   rR   rS   rX   rY   rZ   r^   rb   rc   rd   re   rf   ri   rl   rm   rn   ro   rr   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )returnr   (K  
__future__r   	importlibr   r   __annotations__r   r   __all__numpyr6  r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   
dask.arrayr'   r(   r)   r*   r+   r,   r-   dask.array._shuffler.   dask.array.blockwiser/   r0   dask.array.chunk_typesr1   dask.array.corer2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   dask.array.creationrE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   dask.array.gufuncrU   rV   rW   dask.array.numpy_compatrX   rY   dask.array.optimizationrZ   dask.array.overlapr[   r\   dask.array.percentiler]   r^   dask.array.rechunkr_   dask.array.reductionsr`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   dask.array.reshaper   r   dask.array.routinesr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   dask.array.tiledb_ior   r   dask.array.ufuncr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r	  r
  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  r*  r+  r,  r-  r.  r/  dask.array.utilsr0  dask.array.wrapr1  r2  r3  r4  	dask.baser5  dask.array._array_expr_array_exprdareloadImportErrormsgstrr>  r?  r@  r   r   r   <module>rb     s   "   #' K &I.!
dL	l4#    * KJJ+4:     *    $ BA:04?*" " " " " " " " "F >J J J J J J J J J J J J J J J J J J JV <c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c c cH +*888!!++Yb! U"0>~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
 ~	
~	
 	
 	
 	
 	
* /z:)%0)%0,X6(.*629:OP+G467IJ2>B2>B45EF/<	+G4-i82>B-i82>B*62.z:)%02>B-i8.z:)%0*62)%0.z:.z:.z:0>
,X6,X6-i8-i8,X6,X6/<	/<	0>
/<	/<	*62+G4-i8.z:,X667IJ56GH.z:,X60>
0>
0>
-i8.z:,X6-i8.z:.z:3OD)%0,X6*62.z:)%0,X6-i8,X61-@-i81-@*62,X6,X6.z:/<	1-@1-@,X6,X6-i8*62,X6,X6*62-i8-i8+G4/<	)%0+G478KL1-@*62+G4+G42>B,X6+G4-i8.z:*62/<	/<	*622>B78KL*622>B78KL-i8,X63OD*62,X6+G41-@/<	..%%u   4	V  c!fvo+
,!34L	  "Yb!"s+   Me& 3W1f &f,ff"f43f4