o Nrf @sddlmZddlZddlZddlmZddlmZeej Z e j dkZ e j dkZ e j dkZe j dkZe j d kZerKdd lmZmZdd lmZmZndd lmZmZdd lmZdd lmZzJe<eejddddejdddedreejdddddreedddsedejZejjZWdn1swYWneyd#ddZejj eejj!ddZYnwGdddZ"eeddZ#d$ddZ$d%d!d"Z%dS)&) annotationsN)Version) derived_from))r)r)r)r) AxisErrorComplexWarning)normalize_axis_indexnormalize_axis_tuple)r )rg?runsafe)casting)rdtypeg?i8)rr zIDivide not working with dtype: https://github.com/numpy/numpy/issues/3484cCs$t|||}|dur||}|S)zImplementation of numpy.divide that works with dtype kwarg. Temporary compatibility fix for a bug in numpy's version. See https://github.com/numpy/numpy/issues/3484 for the relevant issue.N)npdivideZastype)x1Zx2outrxr`/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/dask/array/numpy_compat.pyr0s rc@s>eZdZdZddZddddddfdd Zdd d Zd S) _Recurserz; Utility class for recursing over nested iterables cCs ||_dSN recurse_if)selfrrrr__init__Hs z_Recurser.__init__cK|SrrrkwargsrrrNz_Recurser.cKr rrr!rrrr#Or$cKr rr)r"rrrr#Pr$c s$fdd|fi|S)a{ Iterate over the nested list, applying: * ``f_map`` (T -> U) to items * ``f_reduce`` (Iterable[U] -> U) to mapped items For instance, ``map_reduce([[1, 2], 3, 4])`` is:: f_reduce([ f_reduce([ f_map(1), f_map(2) ]), f_map(3), f_map(4) ]]) State can be passed down through the calls with `f_kwargs`, to iterables of mapped items. When kwargs are passed, as in ``map_reduce([[1, 2], 3, 4], **kw)``, this becomes:: kw1 = f_kwargs(**kw) kw2 = f_kwargs(**kw1) f_reduce([ f_reduce([ f_map(1), **kw2) f_map(2, **kw2) ], **kw1), f_map(3, **kw1), f_map(4, **kw1) ]], **kw) csH|s |fi|Sdi|fdd|Dfi|S)Nc3s |] }|fiVqdSrr).0xi)f next_kwargsrr ysz2_Recurser.map_reduce..f..rrr!r'f_kwargsf_mapf_reducer)r(rr'ts  z_Recurser.map_reduce..fr)rrr,r-r+r"rr*r map_reduceKs)z_Recurser.map_reducerccsN||}|||fV|sdSt|D]\}}||||fEdHqdS)aB Iterate over x, yielding (index, value, entering), where * ``index``: a tuple of indices up to this point * ``value``: equal to ``x[index[0]][...][index[-1]]``. On the first iteration, is ``x`` itself * ``entering``: bool. The result of ``recurse_if(value)`` N)r enumeratewalk)rrindexZ do_recurseir&rrrr0}s z_Recurser.walkN)r)__name__ __module__ __qualname____doc__rr.r0rrrrr?s 2rcs~t|jdt||jd}tt|krtdfddt|jD}tt|D] \}}|||q-||}|S)Nsource destinationzJ`source` and `destination` arguments must have the same number of elementscsg|]}|vr|qSrr)r%nr7rr szmoveaxis..) rndimlen ValueErrorrangesortedzipinsert transpose)ar7r8orderdestsrcresultrr:rmoveaxiss rIcCs|j}t||}|dkr||7}d}d|kr|dks-nt|d| d|d|f||kr5|d8}||kr=|dSttd|}|||||||S)Nrz5'%s' arg requires %d <= %s < %d, but %d was passed inrstart.)r<r r>listr?removerBrC)rDZaxisrJr9msgZaxesrrrrollaxiss    rNlinearcCs$tr tj|||dStj|||dS)N)method) interpolation) NUMPY_GE_122r percentile)rDqrPrrrrSsrS)NN)r)rO)& __future__rwarningsnumpyrZpackaging.versionrZ dask.utilsr __version__Z _np_versionreleaserRZ NUMPY_GE_123Z NUMPY_GE_124Z NUMPY_GE_125Z NUMPY_GE_200Znumpy.exceptionsr r Znumpy.lib.array_utilsr rZnumpy.core.numericcatch_warningsZallcloserfloat TypeErrormaZ ma_dividecoreZ_DomainedBinaryOperationZ_DomainSafeDividerrIrNrSrrrrs\                S