biUddlmZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddl ZddlmZmZmZddlmZdd lmZmZmZmZmZmZdd lmZdd lmZdd l m!Z!dd l"m#Z#ddl$m%Z%m&Z&m'Z'GddZ(dJdZ)GddZ*dZ+dKdZ,dZ-dZ.dZ/dZ0dZ1dZ2dddddZ3 iZ4d e5d!<e Z6d"Z7 e7d#Z8e7d$Z9e7d%Z:e7d&Z;e7d'Z<e7d(Z=e7d)Z>e7d*Z?e7d+Z@e7d,ZAe7d-ZBe7d.ZCe7d/ZDe7d0ZEe7d1ZFe7d2ZGe7d3ZHe7d4ZIe7d5ZJe7d6ZKe7d7ZLe7d8ZMe7d9ZNe7d:ZOe7d;ZPe7d;ZQe7d<ZRe7d=ZSe7d>ZTe7d?ZUe7d@ZVe7dAZWe7dBZXe7dCZYe7dDZZe7dEZ[e7dFZ\e7dGZ]e7dHZ^e7dIZ_y)L) annotationsN)chainproduct)Integral)getitem)Lock)TaskTaskRef parse_input)array_creation_dispatch)Arrayasarraybroadcast_shapes broadcast_tonormalize_chunksslices_from_chunks)arange) asarray_safe)tokenize)HighLevelGraph) derived_fromrandom_state_datatypenamec~ eZdZdZdZdZedZedZe e jjdd1d Z e e jjdd1d Ze e jjdd1d Ze e jjd d2d Ze e jjdd3dZe e jjdd1dZe e jjdd3dZe e jjdd1dZe e jjdd4dZe e jjdd1dZe e jjddde j.dd fdZe e jjdd4dZe e jjdd4dZe e jjdd4dZe e jjdd1dZe e jjdd1dZe e jjd d5dZe e jjdd1dZe e jjdd1dZ e e jjdd1dZ!e e jjdd4dZ"e e jjdd1d Z#e e jjdd!Z$e e jjdd3d"Z%e e jjdd1d#Z&e e jjdde jNdd fd$Z e e jjdd3d%Z(e e jjdd1d&Z)e e jjdd1d'Z*e e jjdd1d(Z+e e jjdd1d)Z,e e jjdd1d*Z-e e jjdd1d+Z.e e jjdd4d,Z/e e jjdd1d-Z0e e jjdd1d.Z1e e jjdd1d/Z2e e jjdd1d0Z3y)6 Generatora Container for the BitGenerators. ``Generator`` exposes a number of methods for generating random numbers drawn from a variety of probability distributions and serves as a replacement for ``RandomState``. The main difference between the two is that ``Generator`` relies on an additional ``BitGenerator`` to manage state and generate the random bits, which are then transformed into random values from useful distributions. The default ``BitGenerator`` used by ``Generator`` is ``PCG64``. The ``BitGenerator`` can be changed by passing an instantiated ``BitGenerator`` to ``Generator``. The function :func:`dask.array.random.default_rng` is the recommended way to instantiate a ``Generator``. .. warning:: No Compatibility Guarantee. ``Generator`` does not provide a version compatibility guarantee. In particular, as better algorithms evolve the bit stream may change. Parameters ---------- bit_generator : BitGenerator BitGenerator to use as the core generator. Notes ----- In addition to the distribution-specific arguments, each ``Generator`` method takes a keyword argument `size` that defaults to ``None``. If `size` is ``None``, then a single value is generated and returned. If `size` is an integer, then a 1-D array filled with generated values is returned. If `size` is a tuple, then an array with that shape is filled and returned. The Python stdlib module `random` contains pseudo-random number generator with a number of methods that are similar to the ones available in ``Generator``. It uses Mersenne Twister, and this bit generator can be accessed using ``MT19937``. ``Generator``, besides being Dask-aware, has the advantage that it provides a much larger number of probability distributions to choose from. All ``Generator`` methods are identical to ``np.random.Generator`` except that they also take a `chunks=` keyword argument. ``Generator`` does not guarantee parity in the generated numbers with any third party library. In particular, numbers generated by `Dask` and `NumPy` will differ even if they use the same seed. Examples -------- >>> from numpy.random import PCG64 >>> from dask.array.random import Generator >>> rng = Generator(PCG64()) >>> rng.standard_normal().compute() # doctest: +SKIP array(0.44595957) # random See Also -------- default_rng : Recommended constructor for `Generator`. np.random.Generator c||_yN)_bit_generator)self bit_generators L/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/random.py__init__zGenerator.__init___s +c|jj}|d|jjjzdzz }|S)N()) __class____name__r)r_strs r!__str__zGenerator.__str__bs<~~&& d))33<<kwargss r!r;zGenerator.betas$1Q4Q&QQr#c &t|d||f||d|SNbinomialr<r?rnpr=r>rDs r!rHzGenerator.binomialw$ AqUtFUfUUr#c $t|d|f||d|SN chisquarer<r?rdfr=r>rDs r!rOzGenerator.chisquare{$ RTd6TVTTr#ct|||||||\}}}}}}}} tt|} t|jt | } dt | |||||||z} t| ggd|D} t| | | Dcic]\}}}|t|t|||||||  }}}}tj| || }t|| ||Scc}}}w)Nda.random.choice-%sc3DK|]}tt|ywrrangelen.0bds r! z#Generator.choice..s ABs2w A  dependenciesmeta) _choice_validate_paramslistr_spawn_bitgensrrXrzipr _choice_rngrfrom_collectionsr )rrBr=replacerKaxisshuffler>rar_sizesbitgensnamekeyskbitgendskgraphs r!choicezGenerator.choices( $D!T7AtV L      Wf%& !4!4c%jA$x T61gq$(  vB A& AB$'tWe#<  64 tA{FAtWawO O  //c UUD&t44  s#Cc $t|d|f||d|SN exponentialr<r?rscaler=r>rDs r!rvzGenerator.exponential( - -1& DJ  r#c &t|d||f||d|SNfr<r?rdfnumdfdenr=r>rDs r!r|z Generator.f$UEVVVvVVr#c &t|d||f||d|SNgammar<r?rshaperxr=r>rDs r!rzGenerator.gamma* '5% .26 EK  r#c $t|d|f||d|SN geometricr<r?rrKr=r>rDs r!rzGenerator.geometric$ QST&SFSSr#c &t|d||f||d|SNgumbelr<r?rlocrxr=r>rDs r!rzGenerator.gumbel* (C -1& DJ  r#c (t|d|||f||d|SNhypergeometricr<r?rngoodnbadnsampler=r>rDs r!rzGenerator.hypergeometric5           r#Fc *t|d|f|||||d|S)Nintegers)highr=dtypeendpointr>r?)rlowrr=rrr>rDs r!rzGenerator.integerss:         r#c &t|d||f||d|SNlaplacer<r?rs r!rzGenerator.laplace* )S% .26 EK  r#c &t|d||f||d|SNlogisticr<r?rs r!rzGenerator.logistic* *c5 /3F FL  r#c &t|d||f||d|SN lognormalr<r?rmeansigmar=r>rDs r!rzGenerator.lognormal* +tU 15f HN  r#c $t|d|f||d|SN logseriesr<r?rs r!rzGenerator.logseriesrr#c >t|d||f||t|ffd|SN multinomialr=r> extra_chunksr@rXrrJpvalsr=r>rDs r!rzGenerator.multinomial>     u:-)    r#c (t|d||f|||d|S)Nmultivariate_hypergeometric)r=methodr>r?)rcolorsrr=rr>rDs r!rz%Generator.multivariate_hypergeometrics7  )        r#c &t|d||f||d|SNnegative_binomialr<r?rIs r!rzGenerator.negative_binomial+ %q! 26v IO  r#c &t|d||f||d|SNnoncentral_chisquarer<r?rrQnoncr=r>rDs r!rzGenerator.noncentral_chisquare+ ("d 9=f PV  r#c (t|d|||f||d|SN noncentral_fr<r?rr~rrr=r>rDs r!rzGenerator.noncentral_f , .% ;? RX  r#c &t|d||f||d|SNnormalr<r?rs r!rzGenerator.normal&rr#c $t|d|f||d|SNparetor<r?rrBr=r>rDs r!rzGenerator.pareto,$!P$vPPPr#cddlm}|jdk(r tdt |t j r t|d}|jjt|}t|j||||S)Nr shuffle_slicecupyz`Generator.permutation` not supported for cupy-backed Generator objects. Use the 'numpy' array backend to call `dask.array.random.default_rng`, or pass in `numpy.random.PCG64()`.r8r>) dask.array.slicingrr0NotImplementedError isinstancenumbersNumberrr4rX_shufflerrxrindexs r! permutationzGenerator.permutation0sv4    '%+  a (q(A $$SV,$$e,Q&&r#c $t|d|f||d|SNpoissonr<r?rlamr=r>rDs r!rzGenerator.poissonC$ 3ST&SFSSr#c $t|d|f||d|SNpowerr<r?rs r!rzGenerator.powerG$OfOOOr#c &t|df||||d|S)Nrandom)r=routr>r?)rr=rrr>rDs r!rzGenerator.randomKs* ( !%UF NT  r#c $t|d|f||d|SNrayleighr<r?rws r!rzGenerator.rayleighQ$ EVVVvVVr#c "t|df||d|SNstandard_cauchyr<r?rr=r>rDs r!rzGenerator.standard_cauchyU$ 1VVVvVVr#c "t|df||d|SNstandard_exponentialr<r?rs r!rzGenerator.standard_exponentialY' ( /3F FL  r#c $t|d|f||d|SNstandard_gammar<r?rrr=r>rDs r!rzGenerator.standard_gamma_) "E 04V GM  r#c "t|df||d|SNstandard_normalr<r?rs r!rzGenerator.standard_normalerr#c $t|d|f||d|SN standard_tr<r?rPs r!r zGenerator.standard_ti$ bUtFUfUUr#c (t|d|||f||d|SN triangularr<r?rleftmoderightr=r>rDs r!r zGenerator.triangularm, ,dE 8rDs r!rzGenerator.uniforms* )S$ -1& DJ  r#c &t|d||f||d|SNvonmisesr<r?rmukappar=r>rDs r!rzGenerator.vonmisesy* *b% .26 EK  r#c &t|d||f||d|SNwaldr<r?rrrxr=r>rDs r!r!zGenerator.wald $eX$vXQWXXr#c $t|d|f||d|SNweibullr<r?rs r!r&zGenerator.weibull$ 1Q4Q&QQr#c $t|d|f||d|SNzipfr<r?rs r!r*zGenerator.zipf$NVNvNNr#Nr8)NTNrTr8?Nr8gr.Nr8)N marginalsr8)4r( __module__ __qualname____doc__r"r*propertyr0r4rnprrr;rHrOrsrvr|rrrrint64rrrrrrrrrrrrrrrfloat64rrrrrr r rrr!r&r*r#r!rrs>@, ;; ;; "))%%!4R5R"))%%!4V5V"))%%!4U5U"))%%!4  "55"5H"))%%!4 5 "))%%!4W5W"))%%!4 5 "))%%!4T5T"))%%!4 5 "))%%!4  5  "))%%!4 hh 5 ,"))%%!4 5 "))%%!4 5 "))%%!4 5 "))%%!4T5T"))%%!4  5  "))%%!4EK  5  "))%%!4 5 "))%%!4 5 "))%%!4 5 "))%%!4 5 "))%%!4Q5Q"))%%!4'5'$"))%%!4T5T"))%%!4P5P"))%%!4bjjd6 5 "))%%!4W5W"))%%!4W5W"))%%!4 5 "))%%!4 5 "))%%!4W5W"))%%!4V5V"))%%!4 5 "))%%!4 5 "))%%!4 5 "))%%!4Y5Y"))%%!4R5R"))%%!4O5Or#rct|dr t|St|tr|St|drt|jStt j |S)af Construct a new Generator with the default BitGenerator (PCG64). Parameters ---------- seed : {None, int, array_like[ints], SeedSequence, BitGenerator, Generator}, optional A seed to initialize the `BitGenerator`. If None, then fresh, unpredictable entropy will be pulled from the OS. If an ``int`` or ``array_like[ints]`` is passed, then it will be passed to `SeedSequence` to derive the initial `BitGenerator` state. One may also pass in a `SeedSequence` instance. Additionally, when passed a `BitGenerator`, it will be wrapped by `Generator`. If passed a `Generator`, it will be returned unaltered. Returns ------- Generator The initialized generator object. Notes ----- If ``seed`` is not a `BitGenerator` or a `Generator`, a new `BitGenerator` is instantiated. This function does not manage a default global instance. Examples -------- ``default_rng`` is the recommended constructor for the random number class ``Generator``. Here are several ways we can construct a random number generator using ``default_rng`` and the ``Generator`` class. Here we use ``default_rng`` to generate a random float: >>> import dask.array as da >>> rng = da.random.default_rng(12345) >>> print(rng) Generator(PCG64) >>> rfloat = rng.random().compute() >>> rfloat array(0.86999885) >>> type(rfloat) Here we use ``default_rng`` to generate 3 random integers between 0 (inclusive) and 10 (exclusive): >>> import dask.array as da >>> rng = da.random.default_rng(12345) >>> rints = rng.integers(low=0, high=10, size=3).compute() >>> rints array([2, 8, 7]) >>> type(rints[0]) Here we specify a seed so that we have reproducible results: >>> import dask.array as da >>> rng = da.random.default_rng(seed=42) >>> print(rng) Generator(PCG64) >>> arr1 = rng.random((3, 3)).compute() >>> arr1 array([[0.91674416, 0.91098667, 0.8765925 ], [0.30931841, 0.95465607, 0.17509458], [0.99662814, 0.75203348, 0.15038118]]) If we exit and restart our Python interpreter, we'll see that we generate the same random numbers again: >>> import dask.array as da >>> rng = da.random.default_rng(seed=42) >>> arr2 = rng.random((3, 3)).compute() >>> arr2 array([[0.91674416, 0.91098667, 0.8765925 ], [0.30931841, 0.95465607, 0.17509458], [0.99662814, 0.75203348, 0.15038118]]) See Also -------- np.random.default_rng capsuler )hasattrrrr r default_bit_generator)seeds r! default_rngr>sZdtY D) $  '++,, ,BB4H IIr#c eZdZdZd/dZedZd0dZee jjdd1dZ ee jjdd1d Z ee jjdd1d Zej e5ee jjdd2d Zdddee jjdd3d Zee jjdd1d Zee jjdd3dZee jjdd1dZee jjdd4dZee jjdd1dZee jjdd4dZee jjdd4dZee jjdd4dZee jjdd1dZee jjdd1dZee jjdd1dZee jjdd1dZee jjdd1dZ ee jjdd4dZ!ee jjdd1dZ"ee jjddZ#ee jjdd3dZ$ee jjdd1dZ%ee jjdd5dZ&ee jjdd6d Z'ee jjdd1d!Z(e(Z ee jjdd3d"Z)ee jjdd1d#Z*ee jjdd1d$Z+ee jjdd1d%Z,ee jjdd1d&Z-ee jjdd1d'Z.ee jjdd1d(Z/ee jjdd1d)Z0ee jjdd4d*Z1ee jjdd1d+Z2ee jjdd1d,Z3ee jjdd1d-Z4ee jjdd1d.Z5y#1swYxYw)7 RandomStatea Mersenne Twister pseudo-random number generator This object contains state to deterministically generate pseudo-random numbers from a variety of probability distributions. It is identical to ``np.random.RandomState`` except that all functions also take a ``chunks=`` keyword argument. Parameters ---------- seed: Number Object to pass to RandomState to serve as deterministic seed RandomState: Callable[seed] -> RandomState A callable that, when provided with a ``seed`` keyword provides an object that operates identically to ``np.random.RandomState`` (the default). This might also be a function that returns a ``mkl_random``, or ``cupy.random.RandomState`` object. Examples -------- >>> import dask.array as da >>> state = da.random.RandomState(1234) # a seed >>> x = state.normal(10, 0.1, size=3, chunks=(2,)) >>> x.compute() array([10.01867852, 10.04812289, 9.89649746]) See Also -------- np.random.RandomState Nctjj||_|tj|_y||_yr)r5rr@ _numpy_stater _RandomState)rr=r@s r!r"zRandomState.__init__ s;II11$73>3F # / / LW r#czt|jjdd}tj|Sr,)rrCr.r2r3)rr0s r!r4zRandomState._backends5!!2!2399#>qA &&}55r#c:|jj|yr)rBr=)rr=s r!r=zRandomState.seeds t$r#r5r6c &t|d||f||d|Sr:r?rAs r!r;zRandomState.betarEr#c &t|d||f||d|SrGr?rIs r!rHzRandomState.binomialrLr#c $t|d|f||d|SrNr?rPs r!rOzRandomState.chisquare#rRr#ct|||||d|\}}}}}}}}tt|} tt | |j } dt | |||||z} t| ggd|D} t| | | D cic]\} }}| t| t|||||}}} }tj| ||}t|| ||Scc}}} w)NrrTc3DK|]}tt|ywrrVrYs r!r\z%RandomState.choice..<s$EU3r7^$Er]r^r`) rbrcrrrXrBrrer _choice_rsrrgr )rrBr=rhrKr>rirar_rk state_datarmrnrostaterqrrs r!rszRandomState.choice)s (aw1fM &)*E*3u:t7H7HIJ(8D&!Wa,DD6F$Ef$EFD'*$ E&B"Aud4:uawBBC #33c Ef48 8s!Cc $t|d|f||d|Srur?rws r!rvzRandomState.exponentialGryr#c &t|d||f||d|Sr{r?r}s r!r|z RandomState.fMrr#c &t|d||f||d|Srr?rs r!rzRandomState.gammaQrr#c $t|d|f||d|Srr?rs r!rzRandomState.geometricWrr#c &t|d||f||d|Srr?rs r!rzRandomState.gumbel[rr#c (t|d|||f||d|Srr?rs r!rzRandomState.hypergeometricarr#c &t|d||f||d|Srr?rs r!rzRandomState.laplacenrr#c &t|d||f||d|Srr?rs r!rzRandomState.logistictrr#c &t|d||f||d|Srr?rs r!rzRandomState.lognormalzrr#c $t|d|f||d|Srr?rs r!rzRandomState.logseriesrr#c >t|d||f||t|ffd|Srrrs r!rzRandomState.multinomialrr#c &t|d||f||d|Srr?rIs r!rzRandomState.negative_binomialrr#c &t|d||f||d|Srr?rs r!rz RandomState.noncentral_chisquarerr#c (t|d|||f||d|Srr?rs r!rzRandomState.noncentral_frr#c &t|d||f||d|Srr?rs r!rzRandomState.normalrr#c $t|d|f||d|Srr?rs r!rzRandomState.paretorr#cddlm}t|tjr t |d}t j t|}|jj||||S)Nrrr8r) rrrrrrr5rXrBrjrs r!rzRandomState.permutationsR4 a (q(A #a&! !!%(Q&&r#c $t|d|f||d|Srr?rs r!rzRandomState.poissonrr#c $t|d|f||d|Srr?rs r!rzRandomState.powerrr#c (t|d||f|||d|S)Nrandint)r=r>rr?)rrrr=r>rrDs r!rbzRandomState.randints, )S$ -1& QW  r#c &t|d||f||d|S)Nrandom_integersr<r?rs r!rdzRandomState.random_integerss+ #S$ 59& LR  r#c "t|df||d|S)N random_sampler<r?rs r!rfzRandomState.random_samples$Td6TVTTr#c $t|d|f||d|Srr?rws r!rzRandomState.rayleighrr#c "t|df||d|Srr?rs r!rzRandomState.standard_cauchyrr#c "t|df||d|Srr?rs r!rz RandomState.standard_exponentialrr#c $t|d|f||d|Srr?rs r!rzRandomState.standard_gammarr#c "t|df||d|Srr?rs r!rzRandomState.standard_normalrr#c $t|d|f||d|Srr?rPs r!r zRandomState.standard_tr r#c "t|df||d|S)Ntomaxintr<r?rs r!rnzRandomState.tomaxints$ OfOOOr#c (t|d|||f||d|Sr r?rs r!r zRandomState.triangularrr#c &t|d||f||d|Srr?rs r!rzRandomState.uniformrr#c &t|d||f||d|Srr?rs r!rzRandomState.vonmisesrr#c &t|d||f||d|Sr r?r"s r!r!zRandomState.waldr#r#c $t|d|f||d|Sr%r?rs r!r&zRandomState.weibullr'r#c $t|d|f||d|Sr)r?rs r!r*zRandomState.zipf r+r#)NNrr,)NTNr8r-r/)NNr8l)NNr8)6r(r1r2r3r"r4r4r=rr5rr@r;rHrO contextlibsuppressAttributeErrorrsrvr|rrrrrrrrrrrrrrrrrrbrdrfrrrrrr rnr rrr!r&r*r8r#r!r@r@s> 66 %"))''A6R7R"))''A6V7V"))''A6U7U   ^ ,9 bii++ : 9 ; 99@"))''A6 7 "))''A6W7W"))''A6 7 "))''A6T7T"))''A6 7 "))''A6  7  "))''A6 7 "))''A6 7 "))''A6 7 "))''A6T7T"))''A6  7  "))''A6 7 "))''A6 7 "))''A6 7 "))''A6 7 "))''A6Q7Q"))''A6'7'"))''A6T7T"))''A6P7P"))''A6 7 "))''A6 7 "))''A6U7UF"))''A6W7W"))''A6W7W"))''A6 7 "))''A6 7 "))''A6W7W"))''A6V7V"))''A6P7P"))''A6 7 "))''A6 7 "))''A6 7 "))''A6Y7Y"))''A6R7R"))''A6O7OM99s &WW!r@ct|jdd}tj|}|jj |Sr,)rr.r2r3rr>)rp backend_name backend_libs r!_rng_from_bitgenr|sEF#))#.q1L)),7K    ) )& 11r#c>t|}|j||S)N)ri)r|rj)r rrirMs r!rrs ] +E === &&r#c|jj|}|Dcgc]}t||}}|Scc}wr) _seed_seqspawntype)rp n_bitgensseedsr=rls r!rdrds@    " "9 -E.34d|tF|D!4G4 N5s<ct|tjjr||}t |}t ||}||d|i|S)z$Apply random module method with seedr=)rr5r SeedSequencer|getattr)rngfuncnamerpr=argsrDfuncs r!_apply_random_funcr$sJ&"))001V 6 "C 3 !D  +D +F ++r#cb|tj}||}t||}||d|i|S)z"Apply RandomState method with seedr=)r r@r)r@rrLr=rrDrMrs r! _apply_randomr-s>-99  #E 5( #D  +D +F ++r#cFt|}|j||||||S)N)r=rhrKrirj)r|rs)rLrBr=rhrKrirjrMs r!rfrf6s' Z (E <<gw< WWr#cVtj|}|j||||S)N)r=rhrK)r r@rs)rLrBr=rhrKrMs r!rKrK;s) # / / ;E <<g< ;;r#cg}t|trt|trQ|jdk(r t d|j j jjddd}nrt|trW||j jdgn|} |j j jjdd| }n td|} |dkrtdt|}|j|j}|j}|j dk7r td t#|} |j%|t'|j)d}|t|t*sKt-|| }t/j0|j3dd d s td t|}n|j|j}|j dk7r tdt#|| k7r td|j%|t'|j)d}|d}nt|t4t6fs|f}|dk7r tdt9||t.j:}|st#|ddkDr d} t | ||||||||fS)Nrz3`choice` not supported for cupy-backed `Generator`.r5r8)r=rKzUnknown generator classrza must be greater than 0za must be one dimensional)likegHz>)rtolatolzprobabilities do not sum to 1zp must be one dimensionalza and p must have the same sizez)axis must be 0 since a is one dimensionalrz]replace=False is not currently supported for dask.array.choice with multi-chunk output arrays)rrrr0rr4rr>rsr@array ValueErrorrrechunkr_metandimrXappendr __dask_keys__r rr5isclosesumtuplercrr7) rMrBr=rhrKrir>r_radummy_plen_aerr_msgs r!rbrb@swL!X eY '""f,)I>>((446==abD=QD { +45=enn**A3/aG>>((446==abG=TD67 7 q578 8 AJ IIagg ww 66Q;89 9AA AOO%a( ) }!U#QQ'A::aeegqt!< !@AA A !''"A 66Q;89 9 q6U?>? ?A AOO%a( ) | udm ,w qyDEE fd"** =F s6!9~)   "'** dGQfdL @@r#r8r8rc N |t|ttfs|f}tt||j Dchc].}t|t t jfr |j0c}}||j|t|}t|||jdt j}t|} d} i} i} g} g}t|D]\}}t|t jt fr| |||}t|t r!|j||j | |<n2t|t jrdt#|}|| |<|| |<| j|td|jD| j|i}|j%D]\}}t|t jt fr| |||}t|t r!|j||j | |<n2t|t jrdt#|}|| |<|| |<|td|jD||<|||<tt'|}t|t(r_t+|j,t/|}t#|}|Dcgc]}|j0}}t2}t5|j,}nXt|t6r=t9t/||j:}t#|}t<}|j>}n tAdt#|||||}|d|}t'|gg|Dcgc]}tCt/|c}d ggt/|zz}t'|Dcgc]}tCt/|c}}tE|||| |D]f\}}}}} g}!t|D]\}}|| vr|!j|t|t r"|!jtG| |f| zNt|t jr@tId t#| ||tJtG| ||}"|!j|"tAd i}#|j%D]`\}$}|$| vr||#|$<t|t r | |$f| z|#|$<-t|t jrtJ| |$|f|#|$<WtAd tI||||||tM|!tM|#| |<i|||d t/|z| |}%tOjP|| |}&t |&|||z|%Scc}wcc}wcc}wcc}w)zWrap numpy random function to produce dask.array random function extra_chunks should be a chunks tuple to append to the end of chunks rrct|trt||j|St|tj r)t j t j||Std)Nz!Unknown object type for broadcast)rr rrr5ndarrayascontiguousarray TypeError)arrr>s r!_broadcast_anyz"_wrap_func.._broadcast_anysZ b% E*226: : BJJ '''E(BC C?@ @r#zarray-c3 K|]}dywrNr8rZ_s r!r\z_wrap_func..s&;Qq&; c3 K|]}dywrr8rs r!r\z_wrap_func..s(=q(=rz:Unknown object type: Not a Generator and Not a RandomState-rzgetitem-zUnknown object type in argszUnknown object type in kwargsrr^r`))rrrcrvaluesr r5rrrrrgetr7r enumeratermritemsrrrdrrXrrrr@rrBrrCrrWrer r rr rrg)'rrr=r>rrrDrshapesslicesrrqlookup small_argsr_iresrm small_kwargskeyrkrl bitgen_token_bitgen func_appliergentokenr[rnblocksrpslcblockargt_kwrgrorarrs' r!r@r@sb   4% ?w D&--/2 "ubjj12 HH F  d V $D  jj"**-F   'FA C FJL4 "2 b2::u- . T62C#u%##C(HHq C, / q D   b&;"((&;!;< =   b ! "L<<> #R b2::u- . T62C#u%##C(!hhs C, /"s D "5(=BHH(=#= >L  "L  # &! "E#y! !3!3SZ@( 4;<7$$<<) 3%%& C %#CJ0@0@A( $ TUU \4v >EZq D   ,23b5R>3sec,>O6OO D7"uSW~7 8F),T7E66)R& %VT3t_ CEAr 2b%(JJwq |e';<=BJJ/"8F1Is#;"<=q * B JJrN#$ABB C \\^ EEArQb%(%aylU2DGBJJ/&q 37DG#$CDD E            C;& P   s4y  D  + +D#L QE f|3$ ??I @=47s3VV=V 8V"zdict[str, RandomState]_cached_statescfd}ttj|_ttj|_|S)Nctj}t5 t|}dddt |i|S#t$rt xt|<}Y8wxYw#1swY=xYwr)r backend_cached_states_lockrKeyErrorr@r)rrDrrMattrs r!wrapperz_make_api..wrapper%sr%--  < <&s+ < $wud#T4V44 <.9m;s#e < < r@r|rrdrrrfrKrbr@r__annotations__rrr=r;rHrOrsrvr|rrrrrrrrrrrrrrrrrrfrrbrdrrrrrr r rrr!r&r*r8r#r!rsj"$667').@@kOkO\ \J~cOcOL 2'  ,,X < CAN $FN@b*,&+f  Z  k " 8  &  cN' k " 8 +, I  Z  k " k "  & 12 !78( 8  8  & I '/* ? # I -. Z -. !78+,-. | $ | $ I  Z  I r#