bi5ddlmZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZdd lmZmZe j<eej@fejBej<eej@ejDjFfejHej<eej@fejJe j<eej@fejLe j<eej@fejNe j<ejDjFejDjNe j<eej@fee j<ejDjFeej<ej@d&d Z(e j<ejDjFd'd Z)ej<ejDjFd(d Z*ejVd e jVd ejVd ejVd ejVd ejVd dZ,d'dZ-ejVde jVddZ.ejVde jVdejVdejVdejVddZ/ejVde jVdejVddZ0dZ1ej<ejDjFdZ2ej<eej@fdZ3dZ4d)dZ5ej<eej@fdZ6dZ7GddeZ8ej<ej@dZ9Gd d!e8Z:ed"d#e8d$%Z;e;jyd#e:y)*) annotationsN)chunk)Array) concatenate_lookup divide_lookup einsum_lookup empty_lookupnannumel_lookup numel_lookuppercentile_lookup take_lookuptensordot_lookupto_cupy_dispatchto_numpy_dispatch)divide) ma_divide _percentile)CreationDispatchDaskBackendEntrypointct|||SNr)aqmethods N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/backends.py percentiler%s q!V $$c6tjj||}|Dcgc]}t|ds|j}}t d|Dr t d|r-tj|}t|dk(r |d|_|Scc}w)N)axis fill_valuec3PK|]}t|tj ywr) isinstancenpndarray).0fs r z_concatenate...s ::a $ :s$&zADask doesn't support masked array's with non-scalar `fill_value`sr) r$ma concatenatehasattrr!any ValueErroruniquelen)arraysr outi fill_valuess r _concatenater5*s %%  F  .C)/LA71l3K1<<LKL :k :: O  ii , { q (^CN JMs BBc t||\}} t |}t|} t |}t|}|j }|j }|j } |j } d} ||k7rd} nPt|D]B} ||| | || k7rd} n-|| dkr || xx|z cc<|| dks6|| xx| z cc<D| s tdt|D cgc] } | |vs|  } } | |z}d}|D] }|||z} d|f}| Dcgc]}|| }}t| D cgc] } | |vs|  } } || z}d}|D] }|| |z} |df}| Dcgc]}| | }}|j|j|}|j|j|}tjj||}|j||zS#t$r/tt| d}ttd|}YwxYw#t$r |g}d}YwxYw#t$r |g}d}YwxYwcc} wcc}wcc} wcc}w)Nrr)TFzshape-mismatch for sum) iter TypeErrorlistranger0shapendimr. transposereshaper$r*dot)rbaxesaxes_aaxes_bnanbas_ndabsndbequalknotin newaxes_aN2r newshape_aolda newaxes_b newshape_boldbatbtress r _tensordotrX:s T  [f [f ''C &&C B &&C E Rxr !A6!9~F1I.ay1}q S ay1}q S  ! 122c 61avoQ 6E 6I B c$ibJ"' ($CI (D (c 61avoQ 6E 6I B bhbJ!& 'BtH 'D ' Y  ' ' 3B Y  ' ' 3B %%))B C ;;td{ ##w &eTE1o&eAtn%&    4 7 ) 6 (sW GHH-$ I.I I- I 7I  I4HHH*)H*-H?>H?cupycddltjjjt jjj tjjjtjjttjjttjjttjjfd}t!jt"jfd}t%jjfd}y)Nrc*j|fi|Sr)asnumpydatakwargsrYs r cupy_to_numpyz$register_cupy..cupy_to_numpyt||D+F++rc*j|fi|Sr)asarrayr]s r numpy_to_cupyz$register_cupy..numpy_to_cupyrarcp|jdd|jddj|i|S)Ncastingorder)popeinsum)argsr_rYs r _cupy_einsumz#register_cupy.._cupy_einsums7  9d# 7D!t{{D+F++r)rYrregisterr%r+r taker tensordotr rr _numel_arrayliker _nannumelrrr$r)r`rdrkrYs @r register_cupyrq~s d.>.>?tyy1dllDNN;t||Z8$,,(89T\\95 -,.,rzz*,+,DLL),*,rcb|dvr td|r td|fn |tdf}||S)N>rr)z:Sparse matrices can only be concatenated along axis 0 or 1)r.slice)arrayidxr indexers r sparse_takerws; 6UVV$(uT{C sE$K.@G >rcupyxcddlm} ddlmmdfd }t j ||tj |ttj |ty#t$r}t d|d}~wwxYw)Nr)spmatrix)hstackvstackz>Stacking of sparse arrays requires at least CuPy version 8.0.0cX|dk(r|S|dk(r|Sd|z}t|)Nrr)zECan only concatenate cupy sparse matrices for axis in {0, 1}. Got %s)r.)Lr msgr{r|s r_concat_cupy_sparsez+register_cupyx.._concat_cupy_sparsesC 19!9  QY!9 "$() S/ !rr) cupyx.scipy.sparserzr{r| ImportErrorrrlr_tensordot_scipy_sparser rw)rzerr{r|s @@rregister_cupyxrsn,5 "*=>h(?@;/'  L  sA$$ A>- A99A>sparsecddl}tj|j|jt j|j|j tj|jttj|jttj|jtjy)Nr)rrrlCOOr+rrnr _numel_ndarrayr _nannumel_sparser r$rm)rs rregister_sparsers~  F,>,>?fjj&*:*:;&**n5VZZ)9:RWW-rscipycddldfd }tjjj|t jjjt tjjjt ddlm }tj||t j|t tj|ty#t$rYywxYw)Nrc|dk(rjj|S|dk(rjj|Sd|z}t|)Nrr)zFCan only concatenate scipy sparse matrices for axis in {0, 1}. Got %s)rr|r{r.)r~r rrs rr5z+register_scipy_sparse.._concatenatesY 19<<&&q) ) QY<<&&q) )"$() S/ !r)sparrayr) scipy.sparserrlrrzrrr rwrr)r5rrs @rregister_scipy_sparsers " 5 5|Dell335LM.. <3( ##G\:!!'+BCWk2    sC$$ C0/C0c|j|jcxk(rdk(sJJt|dt|dcxk(rdk(sJJ|d\}|d\}|dvr|dvsJ|j||j|k(sJ|dk(r|dk(r|j|zS|dk(r|dk(r|j|jzS|dk(r |dk(r||zS|dk(r|dk(r||jzSyy)Nrr))rr))r=r0r<T)rrArBa_axisb_axiss rrrs 66QVV q   tAw<3tAw< ,1 ,, ,, ,QIVQIV V & 00 0 776?aggfo -- - {v{ssQw 11ssQSSy 111u 11133w%rc Ttjtj|fi|S)z'Numel implementation for masked arrays.)rsumr$ ones_likexr_s r _numel_maskedrs  99R\\!_ / //rc t|fddi|S)zJNumel implementation for arrays that want to return numel of type ndarray.coerce_np_ndarrayT_numelrs rrr s ! 6t 6v 66rc t|fddi|S)zKNumel implementation for arrays that want to return numel of the same type.rFrrs rroros ! 7u 7 77rc |j|jdd}|jd|jdtj}htj|}|dur|S|r$tj dt z||Stj||dt z|Stttfsgtjfd D}|d ur(tfd tt D}n'tfd tt D}|r+tjtj|||Stj||||S) a  A reduction to count the number of elements. This has an additional kwarg in coerce_np_ndarray, which determines whether to ensure that the resulting array is a numpy.ndarray, or whether we allow it to be other array types via `np.full_like`. keepdimsFr dtype)r)r))r<r!r)r<rc3(K|] }| ywr)r&dimr<s rr(z_numel..0s0CU3Z0sTc34K|]}|vr|ndyw)r)Nrr&rr r<s rr(z_numel..2s% 58#T/E#Jq 0 sc32K|]}|vs|ywrrrs rr(z_numel..6sUSPT_%*Us  )r<getr$float64prodfullr0 full_liker#tupler:mathr; broadcast_tort) rrr_rrr new_shaper r<s @@rrrs? GGEzz*e,H ::f D JJw +E |wwuE* u K 77E !2t5Q Q<<4tc%j/@N N dUDM *v 99040 0D4 s" 99rxx{^ .v ..rc Vt|fi|}t|dr|jS|S)a  A reduction to count the number of elements in a sparse array, excluding nans. This will in general result in a dense matrix with an unpredictable fill value. So make it official and convert it to dense. https://github.com/dask/dask/issues/7169 todense)rpr,r)rr_ns rrrDs/ !vA"!Y/199;6Q6rceZdZdZedZedZeddddZeddddZ eddddZ edddd Z ed dddd Z y) ArrayBackendEntrypointzjDask-Array version of ``DaskBackendEntrypoint`` See Also -------- NumpyBackendEntrypoint ct)zReturn the backend-specific RandomState class For example, the 'numpy' backend simply returns ``numpy.random.RandomState``. NotImplementedErrorselfs r RandomStatez"ArrayBackendEntrypoint.RandomStateZ "!rct)z$Return the default BitGenerator typerrs rdefault_bit_generatorz,ArrayBackendEntrypoint.default_bit_generatorcs "!rN)rmetac t)ztCreate an array of ones Returns a new array having a specified shape and filled with ones. rr<rrr_s roneszArrayBackendEntrypoint.oneshrrc t)zvCreate an array of zeros Returns a new array having a specified shape and filled with zeros. rrs rzeroszArrayBackendEntrypoint.zerosqrrc t)z`Create an empty array Returns an uninitialized array having a specified shape. rrs remptyzArrayBackendEntrypoint.emptyzs "!rc t)zCreate a uniformly filled array Returns a new array having a specified shape and filled with fill_value. r)r<r!rrr_s rrzArrayBackendEntrypoint.fullrrc t)zCreate an ascending or descending array Returns evenly spaced values within the half-open interval ``[start, stop)`` as a one-dimensional array. r)startstopsteprrr_s rarangezArrayBackendEntrypoint.arangerr)Nr)) __name__ __module__ __qualname____doc__propertyrr staticmethodrrrrrrrrrrRs""""!"""""""")-D"""T""rrc |Srr)r^r_s rto_numpy_dispatch_from_numpyrs KrcNeZdZedZeddZedZedZy)NumpyBackendEntrypointctSr)r)clss rto_backend_dispatchz*NumpyBackendEntrypoint.to_backend_dispatchs  rc t|jtjr|S|j|j fi|Sr)r#_metar$r% map_blocksr)rr^r_s r to_backendz!NumpyBackendEntrypoint.to_backends8 djj"** -Kts668CFCCrc6tjjSr)r$randomrrs rrz"NumpyBackendEntrypoint.RandomStatesyy$$$rc6tjjSr)r$rPCG64rs rrz,NumpyBackendEntrypoint.default_bit_generatorsyyrN)r^r) rrr classmethodrrrrrrrrrrsR!!DD %%rrrtnumpyarray_creation_dispatch) module_namedefaultentrypoint_classname)linearr)r)rbool)= __future__rrrr$ dask.arrayrdask.array.corerdask.array.dispatchrrrr r r r r rrrdask.array.numpy_compatr np_dividerdask.array.percentiler dask.backendsrrrlobjectr%r+r* masked_arrayrmrnrirrr5rX register_lazyrqrwrrrrrrrorrprrrrrregister_backendrrrrsP" !    8--AVRZZ0"..A fbjj"%%*<*<=rwwG62::. =  +RYY7 vrzz*BHH5 bee(("%%++6  +Y7 ruu))95BJJ'%(%RUU//0 1 255--.@$/@$F '!!!&)6"v&F#   (,)$'#*( ,4 (!!!'*0+)06 )!!!(+x(H%8$ .%&),* . (!!!'*7#3$+)3:"ruu))*0+0  +,7-7 8 %CP62::.//0/ 7@"2@"FBJJ'(3*+ + " ((2H2JKr