a uDfJ @srddlmZddlZddlZddlZddlZddlZddlZddlZ ddl m Z m Z ddl mZddlmZddlmZmZddlmZddlmZmZmZmZd d Zd:d d Zd dZd;ddZddZddZ ddZ!dddZ$d?d d!Z%ej&fd"d#Z'd$d%Z(d&d'Z)d(d)Z*d*d+Z+d,d-Z,d.d/Z-d0d1Z.d@d2d3Z/d4d5Z0dAd6d7Z1d8d9Z2dS)B) annotationsN)concat frequencies)Array) AxisError)is_dask_collectiontokenize)HighLevelGraph) has_keyword is_arraylike is_cupy_typetypenamecCst|r|S|SdSN)r getxr]/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/dask/array/utils.pynormalize_to_arraysrc st|dr t|r t|r |j}|dur8|dur8tdt|rLt|}|dur\tj}n|durtt|drt|j }t |t r|d|pd|d}t |t st |t rdd |D}d d t||D}t |t r|St |St|d rt|drt |jt s|S|dur|j}z|t d d t|jD}|j|kr||jkr|tft dd t||jD}|t dd t|jD}n"|dkr|}n|d|}|tjjurtjjtjd||p|j ddd}Wn.tytjd||p|j d}Yn0t|r0t|}|r|j |krz||}WnbtyzHtfdd dDr|j jdvrtg|}nWYdn d00|S)aNormalize an array to appropriate meta object Parameters ---------- x: array-like, callable Either an object that looks sufficiently like a Numpy array, or a callable that accepts shape and dtype keywords ndim: int Number of dimensions of the array dtype: Numpy dtype A valid input for ``np.dtype`` Returns ------- array-like with zero elements of the correct dtype _metaNz/You must specify the meta or dtype of the arraydtyperr)shapercSs4g|],}t|tjrdnt|dr(|jnt|qS)rndim) isinstancenumbersNumberhasattrrlen).0arrr @s  z#meta_from_array..cSs&g|]\}}|dkr|nt||qSr)meta_from_array)rr Zndrrrr!Hrcss|]}tdddVqdSrNslicer_rrr Vr#z"meta_from_array..css|] }dVqdSrrr'rrrr)Yr#css|]}tdddVqdSr$r%r'rrrr)Zr#rT)maskc3s|]}|tvVqdSrstrrserrr)ls)zinvalid literalz!could not convert string to floatZSU)rrr r ValueErrornpisscalararrayndarrayrrtypelisttupleziprrrangeEllipsissumZreshapemaZmaskedempty Exceptionastypeanykind)rrrZndimsr metarr0rr"sn       &  *      r"c sDtjdd ttjdtddd|D}dd|D}t|tjrb||}n(z&t |drvd |d<||i|}Wnt yzLt fd d d Drn&WYdWdWddSWYdnd0t y\zXt |d kr"dtvr"|d}n&WYdWdWddSWYdn6d0tyYWdWddS0|rt|dd|krtt||}Wdn1s0Yt|rt|}|WdWdS1s0YWdn1s60YdS)Nignore)all)categorycSs g|]}t|rt|n|qSrr r")rrrrrr!r#z compute_meta..cSs&i|]\}}|t|rt|n|qSrrHrkvrrr sz compute_meta..Zcomputing_metaTc3s|]}|tvVqdSrr,r.r0rrr)szcompute_meta..)zunexpected keyword argumentzis an invalid keyword forz'Did not understand the following kwargsz&zero-size array to reduction operationrr)r3Zerrstatewarningscatch_warnings simplefilterRuntimeWarningitemsrZ vectorizer TypeErrorrBr2rr-r@getattr contextlibsuppressAttributeErrorrAr4r5)funcZ_dtypeargskwargsZ args_metaZ kwargs_metarDrr0r compute_meta|sH    :  :  *  r[FcKst|}t|}t|dddkrft|ds4t|drNtjj||fddi|Stj||fd|i|S|r|j|jkotddt|j |j DS||kS) NrOr+Z masked_equalT equal_nancss.|]&\}}t|rt|n||kVqdSr)r3isnan)rr brrrr)szallclose..) rrTrr3r>allcloserrFr:Zflat)r r_r]rZrrrr`s r`cCs$dd}t|j|dt|j|dkS)NcSst|tr|dddfS|SdS)N)rr-)rJrrrkeys  zsame_keys..key)rb)sorteddask)r r_rbrrr same_keyssrecCs|jod|jvSNr)rrrrr _not_emptysrgc Cst|tsdS|tdd|jDs.Jtt|j}dd|D}t }| D]>}|jD].}z| t ||Wqnt yYqn0qnq`t|dksJ|dS)z2Check that graph is well named and non-overlappingNcss|]}t|ttfVqdSr)rr9r-)rrJrrrr)r#z_check_dsk..cSsi|]\}}|dkr||qS)rMrrIrrrrLr#z_check_dsk..)rr validaterFZlayersrrvaluesrRsetkeysaddrKeyErrorr)ZdskZfreqsZnon_oneZkey_collisionsrJZlayerrrr _check_dsks    roTcCsn|rt|t|ksJt||D]F\}}t|s>t|r\|rht|t|kshJq"||ks"Jq"dSr)rr:mathr^)r r_ check_ndim check_nanZaaZbbrrrassert_eq_shapesrscCs|j|d}tjdd|jDD]}|j|jf|}t|drJ|}t|dsbtj |dd}t ddt |j|D}t ||j |d d |j|jks"Jd q"|S) N schedulercss|]}tt|VqdSr)r;r)rcrrrr)r#z _check_chunks..resultrr\r*css|]\}}||VqdSrr)rrvirrrr)r#F)rqrrz6maybe you forgot to pass the scheduler to `assert_eq`?)Zpersist itertoolsproductchunksrdnamerrwr3r5r9r:rsrr)rrqruidxchunkZexpected_shaperrr _check_chunkss      rc Cs|}d}d}t|rt|r|jdus*J|j} |r>t|jt|dd}|r\t|||d}|j|d}|}t|dr~| }t|dst j |dd}t |r|j|jksJ|rt |j|jdd n$t|dst j |dd}t|dd} || ||fS) Nr)rqrurttodenserr\r*F)rr)rr rrordrTrcomputerrr3r5rgrsr) r check_shape check_graph check_chunksrqruZ x_originalZx_metaZ x_computedadtrrr_get_dt_meta_computeds2       rsyncc KsT|} |} t|tttfr"t|}t|tttfr.)assignedcSs|Srrrrrrr#zsafe_wraps..N)rF functoolswraps)rrrrr safe_wrapssrcCs,z|jWSty&t|jYS0dS)z!Determine dtype of an array-like.N)rrWr3 asanyarray)r rrr _dtype_ofs rcOsZ|durtj|i|Sztj|dt|i|WStyTtj|i|YS0dS)z Use the `like=` from `np.arange` to create a new array dispatching to the downstream library. If that fails, falls back to the default NumPy behavior, resulting in a `numpy.ndarray`. Nlike)r3Zaranger"rS)rrYrZrrr arange_safes  rcKs||urt|dr|St|tr0||fi|St|trPt|jrP|jdd}z||fdt|i|WSty||fi|YS0dS)NZ__array_function__rrtr)rrrr rrr"rS)Znp_funcZda_funcr rrZrrr_array_like_safes     rcKs$ddlm}ttj|||fi|S)a If `a` is `dask.array`, return `dask.array.asarray(a, **kwargs)`, otherwise return `np.asarray(a, like=like, **kwargs)`, dispatching the call to the library that implements the like array. Note that when `a` is a `dask.Array` backed by `cupy.ndarray` but `like` isn't, this function will call `a.compute(scheduler="sync")` before `np.array`, as downstream libraries are unlikely to know how to convert a `dask.Array` and CuPy doesn't implement `__array__` to prevent implicit copies to host. r)r5)Zdask.array.routinesr5rr3)r rrZr5rrr array_safes rcKs$ddlm}ttj|||fi|S)a If a is dask.array, return dask.array.asarray(a, **kwargs), otherwise return np.asarray(a, like=like, **kwargs), dispatching the call to the library that implements the like array. Note that when a is a dask.Array but like isn't, this function will call a.compute(scheduler="sync") before np.asarray, as downstream libraries are unlikely to know how to convert a dask.Array. r)asarray)dask.array.corerrr3)r rrZrrrr asarray_safes rcKs$ddlm}ttj|||fi|S)a If a is dask.array, return dask.array.asanyarray(a, **kwargs), otherwise return np.asanyarray(a, like=like, **kwargs), dispatching the call to the library that implements the like array. Note that when a is a dask.Array but like isn't, this function will call a.compute(scheduler="sync") before np.asanyarray, as downstream libraries are unlikely to know how to convert a dask.Array. r)r)rrrr3)r rrZrrrrasanyarray_safes rcsrt|ttfr$tfdd|DSt|tjs.z%Axis value must be an integer, got %sz2Axis %d is out of bounds for array of dimension %dr)rr9r8rIntegralrSr)axisrrrrrs   rcCsf|r|j}tj|ddd}n|j}tj|dddj}d|dkd|}||||j}}||fS)aSign correction to ensure deterministic output from SVD. This function is useful for orienting eigenvectors such that they all lie in a shared but arbitrary half-space. This makes it possible to ensure that results are equivalent across SVD implementations and random number generator states. Parameters ---------- u : (M, K) array_like Left singular vectors (in columns) v : (K, N) array_like Right singular vectors (in rows) u_based_decision: bool Whether or not to choose signs based on `u` rather than `v`, by default False Returns ------- u : (M, K) array_like Left singular vectors with corrected sign v: (K, N) array_like Right singular vectors with corrected sign rT)rZkeepdimsrMg@g?)rr3r=TrA)urKZu_based_decisionrZsignsrrrsvd_flipsrcOsJ|d}t|r(ddl}t|jj|}nddl}t|j|}||i|Srf)r Zcupyx.scipy.linalgrTscipyZlinalgZ scipy.linalg) func_namerYrZr ZcupyxrXrrrrscipy_linalg_safe*s rcCstd|||dS)NZsolve_triangular)lower)r)r r_rrrrsolve_triangular_safe:srcCs@|dkr(tjdttdtddtStdtd|dS)NrzeAxisError was deprecated after version 2021.10.0 and will be removed in a future release. Please use z instead.rh)rG stacklevelzmodule z has no attribute )rNwarnr r FutureWarningrW__name__)r|rrr __getattr__>sr)NN)F)TT)TN)TTTTN) TTTTTTTTr)F)F)3 __future__rrUrryrprrNnumpyr3ZtlzrrrrZdask.array.numpy_compatrZ dask.baserrZdask.highlevelgraphr Z dask.utilsr r r r rr"r[r`rergrorsrrrWRAPPER_ASSIGNMENTSrrrrrrrrrrrrrrrrsd     a4    * k   *