B T\@sddlmZmZmZddlmZddlmZddlZ yddl m Z Wn e k rdddl m Z YnXddlmZddlmZd d lmZmZd d Ze d dZeeZee jddZee jddZee jddZee jZdS))absolute_importdivisionprint_function)partial)productN)curry)tokenize)funcname)Arraynormalize_chunksc s2d|kr ddd}n |d}t|tjr>|}t|ttfsR|f}|dd}|dd}|dkr|f|j}t|}t|||d}|d d}|pt d t ||||}t |gfd d |D}t |}t fd|i|fd d|D} t t|| } t| |||dS)z= Transform np creation function into blocked version shaperr Nchunksautodtype)rname-cSsg|]}tt|qS)rangelen).0Zbdrr.lib/python3.7/site-packages/dask/array/wrap.py -sz0wrap_func_shape_as_first_arg..c3s|]}f|fVqdS)Nr)rs)argsfuncrr 0sz/wrap_func_shape_as_first_arg..)pop isinstancenpZndarraytolisttuplelistrr r r rrdictzipr ) rrkwargsrrrrkeysZshapesZvalsZdskr)rrrwrap_func_shape_as_first_args*      "r(cKsBt||f|}d}|jdk r>|d|ji|j|_d|j|_|S)Nz Blocked variant of %(name)s Follows the signature of %(name)s exactly except that it also requires a keyword argument chunks=(...) Original signature follows below. rZblocked_)r__doc____name__)Z wrap_funcrr&ftemplaterrrwrap6s   r-Zf8)r)Z __future__rrr functoolsr itertoolsrZnumpyr Zcytoolzr ImportErrorZtoolzbaser Zutilsr Zcorer r r(r-wZonesZzerosemptyZfullrrrrs"    $