tf?ddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZddlmZddlmZmZmZd Zd Zd Zdd Zd Z ddZ ddZddZdZ ddZy)) annotationsN) bisect_left)cycle)add itemgetter) accumulategroupbypluckunique)istask)applyfuncnameimport_requiredc:ddl}ddlm}||jS)Nr)Version)bokehpackaging.versionr __version__)rrs l/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/dask/diagnostics/profile_visualize.py BOKEH_VERSIONrs) 5$$ %%z1Diagnostics plots require `bokeh` to be installedc"t|r|dtttfvr|dt t |dS|dt k(rEt|dtr2t|ddtrt t t |dS|S)Nr)r tuplelistsetmapunquotedict isinstance)exprs rrrs~ d| 7udC( (473wQ01 1 GtO47D)471:t,GT!W-. . Krc2 t|rp|d}|turMt|d}d}t|dkDrt |dnd}t|dkDrt |dni}n^t |dr;dj d |jD}dt|jz}nt|d}d}|dd }i}|s|rAt|t|z t|z t|t|zz  fd  |r# d kDrd j fd|D}nd}nd}|r= d kDr5d d j fdt|jDz}nd}nd}|d|||St|trs|syt|dkDrt|d d|}|d ddzSt|dz dt|zz t|z d j  fd|D}d|dS |vryy#t$rYywxYw)a?Return a nicely formatted string for a task. Parameters ---------- task: Value within dask graph to render as text keys: iterable List of keys within dask graph label_size: int (optional) Maximum size of output label, defaults to 60 Examples -------- >>> from operator import add, mul >>> dsk = {'a': 1, ... 'b': 2, ... 'c': (add, 'a', 'b'), ... 'd': (add, (mul, 'a', 'b'), 'c'), ... 'e': (sum, ['a', 'b', 5]), ... 'f': (add,), ... 'g': []} >>> pprint_task(dsk['c'], dsk) 'add(_, _)' >>> pprint_task(dsk['d'], dsk) 'add(mul(_, _), _)' >>> pprint_task(dsk['e'], dsk) 'sum([_, _, *])' >>> pprint_task(dsk['f'], dsk) 'add()' >>> pprint_task(dsk['g'], dsk) '[]' rr)funcs(c32K|]}t|ywN)r).0fs r zpprint_task..Qs@ @sNct|Sr* pprint_task)tkeys label_size2s rzpprint_task..\s{1dK@rz, c3.K|] }|ywr*r%)r+r1pprints rr-zpprint_task.._s 9q 9sz...c3>K|]\}}|d|yw)=Nr%)r+kvr7s rr-zpprint_task..fs)*+/1aqc6!9+&*sz, ...z[]z, ...]c38K|]}t|ywr*r/)r+r1r2r3s rr-zpprint_task..vsM1[D+>Ms[]_*)r r rlenrhasattrjoinr'intsorteditemsr rr0 TypeError) taskr2 label_sizefuncheadtailargskwargsresultr3r7s ` @@rr0r0&s'Dd|Aw 5=DG$DD'*4y1}747#"D),TQWT!W%BFtW%xx@TZZ@@S_,Q(8DF 6c$i'#d)3TS[9PQKAF Qyy 9D 99D Q *39&,,.3I*!!Fqvhtf-- D$  Y] bq4Bf  +#  ::k* |>>#,,, i &K -6 7TJDJJ  7D 7 4yA~ G1gab A AIAG!A $$q( # B cr A!%AGG "$A $$q( # B   $A "A !#A  $ KKd+!+ ,     xd+  H= 80,s <G0$ G5c Ntdt}ddlm}t ddddd } d |vr|j d |d <d |vr|j d |d <| j d#i||rt|\} } } } }ttd|}|jDcic]-\}}||Dcgc]}|j|jz c}/}}}}tt|jtddDcic] \}}|d| }}}|jd#t!t#|Dcgc] }t%|c}d||z gd| }i}t| | Dcgc] \}}||z  c}}x|d <}t|| Dcgc]\}}|dz |z|z c}}|d<|Dcgc] }||dz c}|d<| Dcgc]}t'|||c}x|d<}t)|||d<| Dcgc] }t%|c}|d<|j+|}|j-|dddd ddn7|jd#t!dDcgc] }t%|c}ddgd| }d|j._d|j2_d|j2_d|j8_d |j<_|j?|}d!|_ d"|_!|Scc}wcc}}}wcc}}wcc}wcc}}wcc}}wcc}wcc}wcc}wcc}w)$aVisualize the results of profiling in a bokeh plot. Parameters ---------- results : sequence Output of Profiler.results dsk : dict The dask graph being profiled. start_time : float Start time of the profile in seconds end_time : float End time of the profile in seconds palette : string, optional Name of the bokeh palette to use, must be a member of bokeh.palettes.all_palettes. label_size: int (optional) Maximum size of output labels in plot, defaults to 60 **kwargs Other keyword arguments, passed to bokeh.figure. These will override all defaults set by visualize. Returns ------- The completed bokeh plot object. rer HoverToolProfile Resultsz!hover,save,reset,xwheel_zoom,xpanabove ,rwtoolstoolbar_locationwidthr{ plot_widthrrzr{rT)keyreversey_rangervr$xyfunctioncolorr)datagray)sourcerrr{rr line_color Nz Worker IDTime (s)as
Key:  @key
Task:  @function
follow_mouser%)"rrT bokeh.modelsrrrrupdaterZr rrHend_time start_time enumeraterGfigurerangerCstrr0rcColumnDataSourcerectgridgrid_line_coloraxisaxis_line_colormajor_tick_line_coloryaxisr}r|selecttooltips point_policy) resultsdskrrr[rKrPrrdefaultsr2tasksstartsendsidsid_groupr;r<itimingsrbr1id_lkrrserwr'rhovers r plot_tasksrs/8 )+= >B&1  Hv **\2w!::m4xHOOf),g&eVT3:a='2DLNNDT  :@1aA15a Q\\)5 5  $w}}JqM4H A aD!G   BII %*3u:%67SV7:-.   585F G6AqQ GGW :=eV:LMAQUQY+MS +./aU1X\/S MR#SK3 $C#SSZ5"7E2W '+,!s1v,U $$$$/    BII VuQx8!s1v81b' VX V!AFF!AFF#'AFF $AGG#AGG HHY E EN(E Ho6  8 !HM/#S,9sH# K50 K0K5 K<L 1LL 8LLLL" 0K5c tdt}ddlm}ddlm}m}tddddd } d |vr@|jd |d <tjd k\rtjdd|vr@|jd|d<tjd k\rtjdd|vr|jd| jd$i||rZt|\} } } |} |}| Dcgc]}|| z  } }|jd$t!dt#| dt!d|| z dd| }ngx} x} } |jd$ddd| }|j$|d}|j'| | |dddd|j(_d|t!| r t-| nd| r t#| nddi|_|j'| | |dddd|j1|dd !d"d#|j2_|Scc}w)%aTPlot resource usage in a bokeh plot. Parameters ---------- results : sequence Output of ResourceProfiler.results start_time : float Start time of the profile in seconds end_time : float End time of the profile in seconds palette : string, optional Name of the bokeh palette to use, must be a member of bokeh.palettes.all_palettes. **kwargs Other keyword arguments, passed to bokeh.figure. These will override all defaults set by plot_resources. Returns ------- The completed bokeh plot object. rer)r\) LinearAxisRange1drzsave,reset,xwheel_zoom,xpanrrrrrrr&/Use width instead of plot_width with Bokeh >= 3rzr{1Use height instead of plot_height with Bokeh >= 3rKdrr)rr)rrrz% CPU)r line_width legend_labelmemoryr$Memory)r y_range_namerrz Memory (MB))rr}rightrr%)rrTrr\rrrrrrrryrorprrZr fix_boundsmaxrUlinerr}rVextra_y_ranges add_layoutr|)rrrr[rPrr\rrrr1memcpuleftrrrcolorss rplot_resourcesrXs4, )+= >B0+  Hv **\2w ? A % MMK L!::m4x ? A % MMM Nv < HOOf7m 3 !!QX ! ! BII q#c(C0q%$,2   C# BII Ch C( C  " "7 +A .FFF  Qi  !AGG' CCQCCS# N A FF  Qi  LLmLgV#AGG HC "s4 H c$|t|||zfS)z6Adjust end point to ensure span of at least `min_span`)r)startendmin_spans rrrs #c58+, ,,rc tdt}ddlm} t ddddd } d |vr@|j d |d <t jd k\rtjd d|vr@|j d|d<t jd k\rtjd| jdi||rtt|d d\} } tt| | z} tfd|}i}|j!D]\}}tj#| d}|D]D}||j$xx|j&z cc<||j(xx|j&zcc<Fdgtt+t,t/dt|j!z||<dg| Dcgc]}||z  c}z} |j0ddd||z gi| }t|j!t3||j5D]?\\}}}|j7dd|d |j9| ||Dcgc]}|c}dAn|j0dddgddgd| }d|d|j:_d|j>_|jA| }d|_!|Scc}wcc}w) a1Visualize the results of profiling in a bokeh plot. Parameters ---------- results : sequence Output of CacheProfiler.results dsk : dict The dask graph being profiled. start_time : float Start time of the profile in seconds end_time : float End time of the profile in seconds metric_name : string Metric used to measure cache size palette : string, optional Name of the bokeh palette to use, must be a member of bokeh.palettes.all_palettes. label_size: int (optional) Maximum size of output labels in plot, defaults to 60 **kwargs Other keyword arguments, passed to bokeh.figure. These will override all defaults set by visualize. Returns ------- The completed bokeh plot object. rerrrz hover,save,reset,wheel_zoom,xpanrrrrrrr&rrzr{rNc$t|dS)Nrr/)drrKs rr4zplot_cache..s;qtS*#Errrvrr)rrlabel)rrrrrz Cache Size (r#rz
Task:  @label
r%)"rrTrrrrrrryrorprrrZrGr r rHfromkeys cache_timemetric free_timerrr rrcr2rrrr}r|rr)rrrr metric_namer[rKrPrrrrrticsgroupsrr;valscntsr<rrrvalrrs ` ` r plot_cachersJ )+= >B&0  Hv **\2w ? A % MMK L!::m4x ? A % MMM NHOOfCM*12. fVd]+,EwO||~ RGAt==q)D .Q\\"ahh."Q[[!QXX-! .cDCq&:N1O!PQQDG  Rsd3a*n33 BII Eq(Z"78 EH E!$TZZ\:gtyy{3S!T  JS# FF **S3EAC3EF    BII Cq"g2w C( C' }A6AGG#AGG HHY EEN H544Fs  J< K)<)z profile.htmlTNN)Viridisr)r) __future__rrWrobisectr itertoolsroperatorrrtlzrr r r dask.corer dask.utilsr rrrrTrr0rcrrrrrr%rrrs{" $2277&I Yx,:DHK ^GIi XV r- ` r