o ni; @sxdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl Z ej dkr4ddlmZnddlmZddlZddl Z ddlmZdd lmZmZd ZeZd Zd ZeZd ZdZdZedededededediZeeeeeeeedZdZ dZ! dZ"dZ#d)ddZ$Gdddej%Z&Gdd d ej%Z'Gd!d"d"e&Z(Gd#d$d$e'Z)Gd%d&d&e&Z*Gd'd(d(e'Z+dS)*z*Terminal, Jupyter and file output for W&B.) annotationsN)CallableIterator) )override)term)ipython sparkline2( CRITICALERRORWARNINGINFODEBUGNOTSET)rFATALrWARNrrrru⢿⣻⣽⣾⣷⣯⣟⡿a a returnPrintercCstrtStS)z8Returns a new printer based on the environment we're in.)r in_jupyter_PrinterJupyter _PrinterTermrrS/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/lib/printer.py new_printeresr!c@sFeZdZdZejejd@ddZejdddAd dZ ej dBdCddZ ejdBdDddZ e dEddZ eejdFddZeejdFddZdGd"d#ZejdHd$d%ZejdHd&d'ZejdBdId)d*ZejdHd+d,ZejdJd.d/ZejdHd0d1ZejdKd3d4ZejdHd5d6ZejdBdLd:d;ZejdMd>d?ZdS)Nrz-An object that shows styled text to the user.rIterator[DynamicText | None]cCdS)aA context manager providing a handle to a block of changeable text. Since `wandb` may be outputting to a terminal, it's important to only use this when `wandb` is performing blocking calls, or else text output by non-`wandb` code may get overwritten. Returns None if dynamic text is not supported, such as if stderr is not a TTY and we're not in a Jupyter notebook. Nrselfrrr dynamic_textpzPrinter.dynamic_textNleveltextstr | list[str] | tuple[str]r)str | int | NoneNonecCr#)zDisplay text to the user. Args: text: The text to display. If given an iterable of strings, they're joined with newlines. level: The logging level, for controlling verbosity. Nrr%r*r)rrr display}r'zPrinter.displaystr percent_done float | NonecCr#)zSet the text on the progress indicator. Args: text: The text to set, which must end with \r. percent_done: The current progress, between 0 and 1. Nrr%r*r1rrr progress_updater'zPrinter.progress_update str | NonecCr#)zClose the progress indicator. After this, `progress_update` should not be used. Args: text: The final text to set on the progress indicator. Ignored in Jupyter notebooks. Nrr%r*rrr progress_closer'zPrinter.progress_close name_or_levelintc Csht|tr zt|WStytd|dtwt|tr'|S|dur-tStd|)zReturns the number corresponding to the logging level. Args: name_or_level: The logging level passed to `display`. Raises: ValueError: if the input is not a valid logging level. zUnknown level name: z, supported levels: NzUnknown status level ) isinstancer0_name_to_levelupperKeyError ValueErrorkeysr9r)r8rrr _sanitize_levels   zPrinter._sanitize_levelboolcCr#)z8Whether text passed to display may contain HTML styling.Nrr$rrr supports_htmlr'zPrinter.supports_htmlcCr#)z=Whether text passed to display may contain arbitrary Unicode.Nrr$rrr supports_unicoder'zPrinter.supports_unicodeserieslist[int | float]cCs|jrt|SdS)zReturns a Unicode art representation of the series of numbers. Also known as "ASCII art", except this uses non-ASCII Unicode characters. Returns None if the output doesn't support Unicode. N)rCr sparkify)r%rDrrr sparkliness zPrinter.sparklinescCr#)z"Returns the text styled like code.Nrr6rrr coder'z Printer.codecCr#)z(Returns the text styled like a run name.Nrr6rrr namer'z Printer.namelinkcCr#)aReturns the text styled like a link. Args: link: The target link. text: The text to show for the link. If not set, or if we're not in an environment that supports clickable links, this is ignored. Nrr%rJr*rrr rJr'z Printer.linkcCr#)z/Returns the text styled to draw less attention.Nrr6rrr secondary_textr'zPrinter.secondary_texttickcCr#)zReturns a frame of an animated loading symbol. May return an empty string. Args: tick: An index into the animation. Nrr%rMrrr loading_symbolr'zPrinter.loading_symbolcCr#)z'Returns the text colored like an error.Nrr6rrr errorr'z Printer.errorrIcCr#)z9Returns the string for a named emoji, or an empty string.Nrr%rIrrr emojir'z Printer.emojicCr#)z)Returns the text styled like a file path.Nrr6rrr filesr'z Printer.filesrowslist[list[str]]titlecCr#)z1Returns a grid of strings with an optional title.Nr)r%rTrVrrr gridr'z Printer.gridcolumns list[str]cCr#)z2Returns the column text combined in a compact way.Nrr%rXrrr panelr'z Printer.panelrr"r*r+r)r,rr-Nr*r0r1r2rr-r*r5rr-)r8r,rr9rrA)rDrErr5r*r0rr0rJr0r*r5rr0rMr9rr0rIr0rr0rTrUrVr5rr0rXrYrr0)__name__ __module__ __qualname____doc__ contextlibcontextmanagerabcabstractmethodr&r/r4r7 staticmethodr@propertyrBrCrGrHrIrJrLrOrPrRrSrWr[rrrr rmsT           c@s eZdZdZejd ddZdS) DynamicTextz5A handle to a block of text that's allowed to change.r*r0rr-cCr#)a^Change the text. Args: text: The text to put in the block, with lines separated by \n characters. The text should not end in \n unless a blank line at the end of the block is desired. May include styled output from methods on the Printer that created this. Nrr6rrr set_textr'zDynamicText.set_textNr*r0rr-)rhrirjrkrnrorsrrrr rr srrcs*eZdZd=fdd Zeejd>ddZedd d?ddZe d@dAddZ ed@dBddZ ed@dCddZ ee dDddZee dDddZedEd d!ZedEd"d#Zed@dFd%d&ZedGd(d)ZedEd*d+ZedHd.d/ZedEd0d1ZedEd2d3Zed@dId7d8ZedJd;d<ZZS)Krrr-csttgd|_dS)N)-\|/)super__init__ itertoolscycle _progressr$ __class__rr rzs z_PrinterTerm.__init__r"ccsVt}|s dVn t|VWddSWddS1s$wYdSr^)rr&_DynamicTermTextr%handlerrr r& s  "z_PrinterTerm.dynamic_textNr(r*r+r)r,cCs.t|ttfr d|n|}|||dSN )r:listtuplejoin_display_fn_mappingr.rrr r/)sz_PrinterTerm.displayCallable[[str], None]cCst|}|tkr tjSt|krtkrtjSt|kr%tkr&tjSt|kr2tkr3tj St |kr?tkrCtj Stj Stj Sr^) rr@rwandb termerrorrrtermwarnrtermlogrr(rrr r3s  z _PrinterTerm._display_fn_mappingr0r1r2cCs"tjt|jd|dddS)N F)newline)rrnextr}r3rrr r4Ds"z_PrinterTerm.progress_updater5cCs|pd}t|dS)NzO )rrr6rrr r7Hsz_PrinterTerm.progress_closerAcCr#)NFrr$rrr rBMz_PrinterTerm.supports_htmlcCstjtjSr^)rutilis_unicode_safesysstderrr$rrr rCRsz_PrinterTerm.supports_unicodecCtj|dd}|S)NT)boldclickstyler%r*retrrr rHWz_PrinterTerm.codecCr)Nyellowfgrrrrr rI\rz_PrinterTerm.namerJcCstj|ddd}|S)NblueT)r underliner)r%rJr*rrrr rJasz_PrinterTerm.linkrIcCs@t}tdkrtjtjrtddddddd}||d S) NWindowsu⭐️u🧹u🚀u🦍u🐢u️⚡)starbroomrocketgorillaturtle lightning) dictplatformsystemrrrrstdoutget)r%rIemojisrrr rRhs z_PrinterTerm.emojicCtj|ddS)Nwhiterrr6rrr rLwsz_PrinterTerm.secondary_textrMr9cCs(|jsdS|tt}tjt|tdS)Nrr)rClen_PROGRESS_SYMBOL_ANIMATIONrr_PROGRESS_SYMBOL_COLOR)r%rMidxrrr rO}s z_PrinterTerm.loading_symbolcCr)Nredrrr6rrr rPz_PrinterTerm.errorcCstj|ddd}|S)NmagentaT)rrrrrrr rSsz_PrinterTerm.filesrTrUrVcshtdd|Ddddt|ddgdfd d |D}|r/|d|dS|dS) Ncss|] }t|dVqdS)rN)r.0rowrrr sz$_PrinterTerm.grid..rz {:>{max_len}}z{}rr rcsg|] }j|diqS)max_lenformatr format_rowrrr sz%_PrinterTerm.grid..)maxrrr%rTrVrWrrr rWs  z_PrinterTerm.gridrXrYcCsdd|Sr)rrZrrr r[rz_PrinterTerm.panelrr-r\r]r^)r)r,rrr_r`rarbrcrerdrfrg)rhrirjrzrrlrmr&r/rprr4r7rqrBrCrHrIrJrRrLrOrPrSrWr[ __classcell__rrr~r rsP        rc@s$eZdZd ddZed d d Zd S)rrterm.DynamicBlockrr-cCs ||_dSr^)_handlerrrr rzs z_DynamicTermText.__init__r*r0cCs|j|dSr^)rrsr6rrr rssz_DynamicTermText.set_textN)rrrr-rtrhrirjrzrrsrrrr rs rcseZdZd;fdd ZeejdddZ ee d>ddZ ed?ddZ ed?ddZ ed@dAddZedBdd Zed?d!d"ZedCd%d&Zed?d'd(Zed?d)d*Ze d@dDd-d.Zed@dEd0d1Zed@dFd5d6ZedGd9d:ZZS)Hrrr-cs*tt|_ddlm}||_dSNr)r/)ryrzr jupyter_progress_barr}IPythonr/_ipython_display)r%r/r~rr rzs    z_PrinterJupyter.__init__r"ccsF|jj|jddd}|rt|VndV||jddS)NrT) display_id)rr/HTML_DynamicJupyterTextupdaterrrr r&s  z_PrinterJupyter.dynamic_textNr(r*r+r)r,cCsVtjr tjjjr dSt|ttfrd|n|}d|}|j |j |dSNz
) rrun _settingssilentr:rrr splitlinesrr/rr.rrr r/s z_PrinterJupyter.displayrAcCr#NTrr$rrr rBrz_PrinterJupyter.supports_htmlcCr#rrr$rrr rCrz _PrinterJupyter.supports_unicoder0cCs d|dS)Nrr6rrr rH z_PrinterJupyter.codecC d|dS)Nz rr6rrr rIrz_PrinterJupyter.namerJr5cCsd|d|p|dS)NzzrrKrrr rJsz_PrinterJupyter.linkrIcCr#NrrrQrrr rRz_PrinterJupyter.emojicCs|Sr^rr6rrr rLrz_PrinterJupyter.secondary_textrMr9cCr#rrrNrrr rOrz_PrinterJupyter.loading_symbolcCr)Nzrrr6rrr rPrz_PrinterJupyter.errorcCr)Nrzrr6rrr rSrz_PrinterJupyter.filesr1r2cCs(|jsdS|dur d}|j||dS)Ng?)r}rr3rrr r4s z_PrinterJupyter.progress_update_cCs|jr |jdSdSr^)r}close)r%rrrr r7sz_PrinterJupyter.progress_closerTrUrVcsddddt|ddgdfdd|D}d|d }|r+d |d |d St|d S) Nrzz {}rzcsg|]}j|qSrrrrrr rsz(_PrinterJupyter.grid..zz
z

z


z
)rr_JUPYTER_TABLE_STYLESrrrr rW s  z_PrinterJupyter.gridrXrYcCs$ddd|D}td|dS)NrcSsg|]}d|dqS)z

r)rcolrrr rsz)_PrinterJupyter.panel..z
r)r_JUPYTER_PANEL_STYLES)r%rXrrrr r[sz_PrinterJupyter.panelrr\r]rarbr^rcrerdr_)rr5rr-rfrg)rhrirjrzrrlrmr&r/rqrBrCrHrIrJrRrLrOrPrSr4r7rWr[rrrr~r rsN          rc@s$eZdZd ddZed ddZd S) rrr-cCsddlm}|j|_||_dSr)rr/r_ipython_to_htmlr)r%rr/rrr rzs  z_DynamicJupyterText.__init__r*r0cCs$d|}|j||dSr)rrrrrr6rrr rs!sz_DynamicJupyterText.set_textNrrtrrrrr rs r)rr),rk __future__rrnrlr{rrtypingrr wandb.utilr version_infortyping_extensionsr wandb.errorsrrr r rrrrrrrr_level_to_namer;rrrrr!ABCrrrrrrrrrrr sn        r