bi<4ddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZddlmZddlmZdd lmZd Zd Zd Zd Z ddZ ddZddZdZ ddZy)) annotationsN) bisect_left)cycle)add itemgetter) accumulategroupbypluckunique)import_optional_dependency)istask)import_requiredcHtd}ddlm}||jS)Nbokehr)Version)r packaging.versionr __version__)rrs ]/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/diagnostics/profile_visualize.py BOKEH_VERSIONrs &w /E) 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-. . Krctdt}tt|}t |}|j |}t|j }|tt||t |dz }||}tt|}t ||kDr$tjdj|tt|t|}|D cgc]} ||  c} Scc} w)aGet a dict mapping funcs to colors from palette. Parameters ---------- palette : string Name of the bokeh palette to use, must be a member of bokeh.palettes.all_palettes. funcs : iterable Iterable of function names zbokeh.palettesr*)r_BOKEH_MISSING_MSGsortedr len all_paletteskeysminrrrandomRandomshufflerzipr) palettefuncspalettes unique_funcsn_funcspalette_lookupr'index color_lookupns r get_colorsr6's/1CDH&-(L,G**73N .%%' (D [w/TQ? @EU#G6'?#G 7|g b!!'*L%.9:L%* +LO ++ +s* C9c >tdt}ddlm}d|vr-t j dt d|jd}||jj }t|ts|g}|Dcgc]}|jdi|} }t| d k(r| d} n| d} | d dD]6} | j| _d| _d | _| xj"d zc_8| dd D]/} d| j$_d | _| xj"d zc_1| D]} d | _d | _|j/| D cgc]} | gc} } |r|j1| |r$|j3|||j5| | Scc}wcc} w)aVisualize the results of profiling in a bokeh plot. If multiple profilers are passed in, the plots are stacked vertically. Parameters ---------- profilers : profiler or list Profiler or list of profilers. filename : string, optional Name of the plot output file. show : boolean, optional If True (default), the plot is opened in a browser. save : boolean, optional If True (default when not in notebook), the plot is saved to disk. mode : str, optional Mode passed to bokeh.output_file() **kwargs Other keyword arguments, passed to bokeh.figure. These will override all defaults set by visualize. Returns ------- The completed bokeh plot object. bokeh.plottingr)state file_pathzrThe file_path keyword argument is deprecated and will be removed in a future release. Please use filename instead.)category stacklevelNrK)mode)rr#bokeh.ior9warningswarn FutureWarningpopcurstatenotebookrr_plotr%x_rangetitlemin_border_topheightxaxis axis_labelmin_border_bottommin_border_leftmin_border_rightgridplotshow output_filesave) profilersfilenamerVrXrBkwargsbpr9proffigsptopfs r visualizerbCs6 )+= >Bf  +#  ::k* |>>#,,, i &K -6 7TJDJJ  7D 7 4yA~ G1gab A AIAG!A  HHNH   cr A!%AGG "$A  HHNH  $A "A !#A  $ KKd+!+ ,     xd+  H1 8$,s <F Fc Jtdt}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. r8r HoverToolProfile Resultsz!hover,save,reset,xwheel_zoom,xpanabove ,rMtoolstoolbar_locationwidthrO plot_widthrm plot_heightrOrT)keyreversey_rangerLr;xyfunctioncolorrq)datagray)sourcerurvrOrmrx line_color Nz Worker IDTime (s)as
Key:  @key
Task:  @function
follow_mouserC)"rr# bokeh.modelsrerrHupdater,r ritemsend_time start_time enumerater$figureranger%strreprr6ColumnDataSourcerectgridgrid_line_coloraxisaxis_line_colormajor_tick_line_coloryaxisrQrPselecttooltips point_policy) resultsdskrrr- label_sizer[r\redefaultsr'tasksstartsendsidsid_groupkvitimingsr5tid_lkr_rysermwr.r{hovers 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 5:#;DG#;;Z5"7E2W '+,!s1v,U $$$$/    BII VuQx8!s1v81b' VX V!AFF!AFF#'AFF $AGG#AGG HHY E EN(E Ho6  8 !HM/#;,9sH# K30 K.K3 K:L 1LL 8LL>LL .K3c 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. r8r)r/) LinearAxisRange1drfzsave,reset,xwheel_zoom,xpanrgrhrirjrnrm/Use width instead of plot_width with Bokeh >= 3rorO1Use height instead of plot_height with Bokeh >= 3rdrrs)rr)rrrpz% CPU)rx line_width legend_labelmemoryr;Memory)rx y_range_namerrz Memory (MB))rrQrightrrC)rr#rr/rrrrrHrmajorrErFrr,r fix_boundsmaxr&linerrQr(extra_y_ranges add_layoutrP)rrrr-r[r\r/rrrrmemcpuleftrrr_colorss rplot_resourcesrs4, )+= >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 rrrPs #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} td|}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. r8rrdrfz hover,save,reset,wheel_zoom,xpanrgrhrirjrnrmrrrorOrNct|S)N)r)ds rzplot_cache..s 47rrrLrurv)rurvlabel)r|rr{r~rsz Cache Size ()rz
Task:  @label
rC)"rr#rrerrHrrrErFrrr,r$r r rfromkeys cache_timemetric free_timerrr rr6r'rrrrQrPrr)rrrr metric_namer-rr[r\rerrrticsgroupsryrvalscntsrrr_rqvalrxrs r plot_cacherUsJ )+= >B&0  Hv **\2w ? A % MMK L!::m4x ? A % MMM NHOOfCM*12. fVd]+,*G4||~ 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 ? J7 J<)z profile.htmlTNN)Viridis<)r) __future__rr)rEbisectr itertoolsroperatorrrtlzrr r r dask._compatibilityr dask.corer dask.utilsrrr#rr6rbrrrrrCrrrst" $22:&&I ,:DHE RGIi XV r- ` r