o NrfJ@sddlmZddlZddlZddlZddlZddlZddlZddlZ 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/d4d5Z0d;d6d7Z1d8d9Z2dS)@) annotationsN)concat frequencies)Array) AxisError)is_dask_collectiontokenize)HighLevelGraph) has_keyword is_arraylike is_cupy_typetypenamecCst|r|S|SN)r getxrY/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/dask/array/utils.pynormalize_to_arraysrc st|drt|rt|r|j}|dur|durtdt|r&t|}|dur.tj}n |dur:t|dr:|j }t |t rI|d|pDd|d}t |t sSt |t rodd |D}d d t||D}t |t rk|St |St|d rt|drt |jt s|S|dur|j}zd|t d d t|jD}|j|kr||jkr|tft dd t||jD}|t dd t|jD}n|dkr|}n|d|}|tjjurtjjtjd||p|j ddd}Wntytjd||p|j d}Ynwt|rt|}|rR|j |krRz||}W|StyQz%tfdd dDrD|j jdvrDtg|}nWYd|Sdww|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|tjr dn t|dr|jnt|qS)rndim) isinstancenumbersNumberhasattrrlen).0arrr @s z#meta_from_array..cSs&g|]\}}|dkr |nt||qSr)meta_from_array)rr ndrrrr!Hs&rcs|] }tdddVqdSrNslicer_rrr Vz"meta_from_array..css|]}dVqdSrrr(rrrr*Yscsr$r%r&r(rrrr*Zr+rT)maskc3|] }|tvVqdSrstrrserrr*l  )zinvalid literalz!could not convert string to floatZSU)rrr r ValueErrornpisscalararrayndarrayrrtypelisttupleziprrrangeEllipsissumZreshapemaZmaskedempty Exceptionastypeanykind)rrrZndimsr metarr3rr"sz      &  $     r"c stjddttjdtddd|D}dd|D}t|tjr.||}nzt |dr8d |d<||i|}Wnut ykzt fd d d DrTWYdWdWddSdwt yz*t |d krdtvr|d}nWYdWdWddSWYdndwtyYWdWddSw|rt|dd|krtt ||}Wdn1swYt|rt|}|WdWdS1swYWddS1swYdS)Nignore)all)categorycSs g|] }t|r t|n|qSrr r")rrrrrr!s z compute_meta..cSs&i|]\}}|t|rt|n|qSrrLrkvrrr sz compute_meta..Zcomputing_metaTc3r.rr/r1r3rrr*r5zcompute_meta..)zunexpected keyword argumentzis an invalid keyword forz'Did not understand the following kwargsz&zero-size array to reduction operationrr)r7Zerrstatewarningscatch_warnings simplefilterRuntimeWarningitemsrZ vectorizer TypeErrorrFr6rr0rDgetattr contextlibsuppressAttributeErrorrEr8r9)funcZ_dtypeargskwargsZ args_metaZ kwargs_metarHrr3r compute_meta|sX         # &    Tr_FcKst|}t|}t|dddkr3t|dst|dr'tjj||fddi|Stj||fd|i|S|rI|j|jkoHtddt|j |j DS||kS) NrOr-Z masked_equalT equal_nancss0|]\}}t|rt|n||kVqdSr)r7isnan)rr brrrr*s  zallclose..) rrXrr7rBallcloserrJr>Zflat)r rcrar^rrrrds  rdcCs$dd}t|j|dt|j|dkS)NcSst|tr |dddfS|S)N)rr0)rNrrrkeys  zsame_keys..key)rf)sorteddask)r rcrfrrr same_keyssricCs|jod|jvSNr)rrrrr _not_emptyrkc Cst|tsdS|tdd|jDsJtt|j}dd|D}t }| D]}|jD]}z | t ||Wq7t yMYq7wq0t|dksYJ|dS)z2Check that graph is well named and non-overlappingNcss|] }t|ttfVqdSr)rr=r0)rrNrrrr*sz_check_dsk..cSsi|] \}}|dkr||qS)rQrrMrrrrPsz_check_dsk..)rr validaterJZlayersrrvaluesrVsetkeysaddrKeyErrorr)ZdskZfreqsZnon_oneZkey_collisionsrNlayerrrr _check_dsks   ruTcCsn|r t|t|ks Jt||D]#\}}t|st|r.|r-t|t|ks-Jq||ks4JqdSr)rr>mathrb)r rc check_ndim check_nanZaaZbbrrrassert_eq_shapesrycCs|j|d}tjdd|jDD]@}|j|jf|}t|dr%|}t|ds1tj |dd}t ddt |j|D}t ||j |d d |j|jksQJd q|S) N schedulercss|] }tt|VqdSr)r?r)rcrrrr*r+z _check_chunks..resultrr`r,css|] \}}||VqdSrr)rr|irrrr*r+F)rwrxz6maybe you forgot to pass the scheduler to `assert_eq`?)Zpersist itertoolsproductchunksrhnamerr}r7r9r=r>ryrr)rrwr{idxchunkZexpected_shaperrr _check_chunkss    rc Cs|}d}d}t|rct|rc|jdusJ|j} |rt|jt|dd}|r.t|||d}|j|d}|}t|dr?| }t|dsKt j |dd}t |rW|j|jksWJ|rbt |j|jdd nt|dsot j |dd}t|dd} || ||fS) Nr)rwr{rztodenserr`r,F)rx)rr rrurhrXrcomputerrr7r9rkryr) r check_shape check_graph check_chunksrwr{Z x_originalZx_metaZ x_computedadtrrr_get_dt_meta_computeds4        rsyncc KsJ|} |} t|tttfrt|}t|tttfrt|}t|||||| d\}}}}t|||||| d\}}}}|rOt|t|krOtd|d|dz/|j |j ksdJd|j d|j d|r|j rk|n| }|j rt|n| }t |t |ksJdt |dt |d|rkt |drt |drt |j|jt | drd| jjd |jd}| jj|jksJ||durd t | jd t |d}t | jt |ksJ|t|st|sd t |d t |d}t |t |ksJ|t | drkd | jjd |jd}| jj|jks$J||durkdt | jd t |d}t | jt |ksEJ|t|skt|skdt |d t |d}t |t |kskJ|d}t||fd| i| s}J|WdStyYnw||k}t|tjr|sJdS|sJdS)N)rrrrwr{z#a and b have different dtypes: (a: z, b: )z"a and b have different shapes (a: z!a and b have different types (a: rz.)assignedcSs|Srrrrrrszsafe_wraps..)rJ functoolswraps)rrrrr safe_wrapssrcCs(z|jWStyt|jYSw)z!Determine dtype of an array-like.)rr[r7 asanyarray)r rrr _dtype_ofs  rcOsV|dur tj|i|Sz tj|dt|i|WSty*tj|i|YSw)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)r7Zaranger"rW)rr]r^rrr arange_safes rcKs||ur t|dr |St|tr||fi|St|tr(t|jr(|jdd}z ||fdt|i|WStyF||fi|YSw)NZ__array_function__rrzr)rrrr rrr"rW)Znp_funcZda_funcr rr^rrr_array_like_safes     rcK$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)r9)Zdask.array.routinesr9rr7)r rr^r9rrr array_safes rcKr)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.corerrr7)r rr^rrrr asarray_safe rcKr)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)rrrr7)r rr^rrrrasanyarray_saferrcsrt|ttfrtfdd|DSt|tjstd|| ks'|kr/td|f|dkr7|7}|S)z#Validate an input to axis= keywordsc3s|]}t|VqdSr) validate_axis)raxrrrr*rz validate_axis..z%Axis value must be an integer, got %sz2Axis %d is out of bounds for array of dimension %dr)rr=r<rIntegralrWr)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)rZkeepdimsrQg@g?)rr7rATrE)urOZu_based_decisionrZsignsrrrsvd_flipsrcOsJ|d}t|rddl}t|jj|}n ddl}t|j|}||i|Srj)r Zcupyx.scipy.linalgrXscipyZlinalgZ scipy.linalg) func_namer]r^r Zcupyxr\rrrrscipy_linalg_safe*s rcCstd|||dS)NZsolve_triangular)lower)r)r rcrrrrsolve_triangular_safe:rlrcCs<|dkrtjdttdtddtStdtd|)NrzeAxisError was deprecated after version 2021.10.0 and will be removed in a future release. Please use z instead.rm)rK stacklevelzmodule z has no attribute )rRwarnr r FutureWarningr[__name__)rrrr __getattr__>sr)NN)F)TT)TN)TTTTN) TTTTTTTTr)3 __future__rrYrrrvrrRnumpyr7ZtlzrrrrZdask.array.numpy_compatrZ dask.baserrZdask.highlevelgraphr Z dask.utilsr r r r rr"r_rdrirkruryrrrWRAPPER_ASSIGNMENTSrrrrrrrrrrrrrrrrsf     a 4    * k   *