biddlmZddlmZddlmZddlZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZd ZdZdZe d"dZeeZe d#dZdZ e jBddee ejDdZ#e jBddee ejHdZ%e jBddee ejLdZ'eeZ(e(ejNejLZ&e jBddee ejRZ*e(ejVejRZ,e*jZBe*jZj]dde*_-e*jZj]dde*_-d Z+d!Z)e*jZe+_-e,jZe)_-y)$) annotationspartial)productN)curry)array_creation_dispatch)Arraynormalize_chunks)meta_from_array)tokenize) blockwise)ArrayChunkShapeDep)funcnamec t|tjr|j}t|tt fs|f}|j dd}|j dd}|j dd}|||g|i|j}tj|}t|||}|xst|dzt||||||z}|||||dS)Nnamechunksautodtyper-)shaperkwargsrr) isinstancenpndarraytolisttuplelistpoprr rr )funcargsrrrrrs J/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/wrap.py_parse_wrap_argsr#s%$  eeT] + ::fd #D ZZ& )F JJw %E }U,T,V,22 HHUOE fe5 9F  8D>C'( eVUD&+D   c d|vr |d|dd}}n|jd}t|tr tdt ||||}|d}|d}|d}|d}|d }t |fd|i|}t tt|x}} t|||t|| i } t| ||||jd d S) = Transform np creation function into blocked version rrNzTDask array input not supported. Please use tuple, list, or a 1D numpy array instead.rrrr) numblocksmeta)rr)) rrr TypeErrorr#rrrangelencore_blockwiserget) r r!rrparsedrrrout_inddep_indgraphs r"wrap_func_shape_as_first_argr30sf1gtABxt 7#% C  dD&% 8F 7OE 7OE H F &>D H F 4 /u / /DeCJ/00Gg   6"  E fE 648P QQr$c <d}t|}|jd|j}t||}|d}|d|d}|d}|d}t |gg|D cgc]} t t | c} } t |} t| } | D cgc]} |} } tt| D] \}}|| |d<fdt| | D}tt| |}t||||jScc} wcc} w) r&rrrrrrc3HK|]\}}tfdi|fzyw)rNr).0ksr!rr s r" z!wrap_func_like..is, T&1aWT , ,! , . 5 Ts")r)) r r.rr#rr+r,r enumeratezipdictr astype)r r!rxr)rr/rrbdkeysshapes_kwir8valsdskrs`` @r"wrap_func_likerGTs" QA 1 D JJw (E dD&% 8F 7OE 7OE H F &>D H F D6 >f=U3r7^= >D f F &\F !Q& !B !$v,'11g TCFO TD s4 C dFU); <<> "s #D  Dc |t||fi|}n t||fi|}d}|j7|d|jiz|jz|_d|jz|_|S)Nz Blocked variant of %(name)s Follows the signature of %(name)s exactly except that it also features optional keyword arguments ``chunks: int, tuple, or dict`` and ``name: str``. Original signature follows below. rblocked_)r__doc____name__) wrap_funcr func_likerftemplates r"wraprPpsp It .v . Iy 3F 3H || 66E $--/ Hr$cX|dk(rdnd}tj||g|d|i||S)Nr'r)r broadcast_to)r rr)r!r null_shapes r"_broadcast_trick_innerrUs6{J ??4HtH:HH% PPr$c`t|}|j|_|j|_|S)a Provide a decorator to wrap common numpy function with a broadcast trick. Dask arrays are currently immutable; thus when we know an array is uniform, we can replace the actual data by a single value and have all elements point to it, thus reducing the size. >>> x = np.broadcast_to(1, (100,100,100)) >>> x.base.nbytes 8 Those array are not only more efficient locally, but dask serialisation is aware of the _real_ size of those array and thus can send them around efficiently and schedule accordingly. Note that those array are read-only and numpy will refuse to assign to them, so should be safe. )rUrJrK)r inners r"broadcast_trickrXs)& #4 (ELLEM]]EN Lr$numpyones)backendrf8rzerosempty)rMfullz*array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z%array([0.1, 0.1, 0.1, 0.1, 0.1, 0.1])z >>> np.full_like(y, [0, 0, 255])zB>>> np.full_like(y, [0, 0, 255]) # doctest: +NORMALIZE_WHITESPACEctj|dk7r"tdt|jd|j d*t |dr|j|d<nt||d<t|||d|S)Nr$fill_value must be scalar. Received instead.r)r fill_value) rndim ValueErrortyperKr.hasattrr_full)rrcr!rs r"r_r_s wwza24 3C3L3L2MY W  zz'" :w '(..F7O":.F7O $e Ef EEr$ctj|dk7r"tdt|jdt |||d|S)Nrrarb)arc)rrdrerfrK _full_like)rjrcr!rs r" full_likerlsZ wwza24 3C3L3L2MY W        r$)N)rR)/ __future__r functoolsr itertoolsrrYrtlzrdask.array.backendsrdask.array.corer r dask.array.utilsr dask.baser dask.blockwiser r- dask.layersr dask.utilsrr#r3rGrPwrUrXregister_inplace ones_likerZ zeros_liker] empty_liker^w_likerlrhr_rkrJreplacerRr$r"rs"73,6*