o -wi? @sndZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl Z ddl m Z ddlZddlZddlmZddlmZdd lmZmZd ZeZd Zd ZeZd ZdZdZedededededediZeeeeeeeedZ dZ! dZ" dZ#dZ$d+d,ddZ%Gd ddej&Z'Gd!d"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) wandb_setup)ipython sparkline2( CRITICALERRORWARNINGINFODEBUGNOTSET)rFATALrWARNrrrru⢿⣻⣽⣾⣷⣯⣟⡿a a settingswandb.Settings | NonereturnPrintercCs2|s t}r |j}trt|dSt|dS)aReturns a printer appropriate for the environment we're in. Args: settings: The settings of a run. If not provided and `wandb.setup()` has been called, then global settings are used. Otherwise, settings (such as silent mode) are ignored. )r)rsingleton_if_setuprr in_jupyter_PrinterJupyter _PrinterTerm)r singletonr"R/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/lib/printer.py new_printeras   r$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. Nr"selfr"r"r# dynamic_textuzPrinter.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. Nr"r(r-r,r"r"r#displayr*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. Nr"r(r-r4r"r"r#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. Nr"r(r-r"r"r#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 ) isinstancer3_name_to_levelupperKeyError ValueErrorkeysr<r)r;r"r"r#_sanitize_levels   zPrinter._sanitize_levelboolcCr&)z8Whether text passed to display may contain HTML styling.Nr"r'r"r"r# supports_htmlr*zPrinter.supports_htmlcCr&)z=Whether text passed to display may contain arbitrary Unicode.Nr"r'r"r"r#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)rFr sparkify)r(rGr"r"r# sparkliness zPrinter.sparklinescCr&)z"Returns the text styled like code.Nr"r9r"r"r#coder*z Printer.codecCr&)z(Returns the text styled like a run name.Nr"r9r"r"r#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. Nr"r(rMr-r"r"r#rMr*z Printer.linkcCr&)z/Returns the text styled to draw less attention.Nr"r9r"r"r#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. Nr"r(rPr"r"r#loading_symbolr*zPrinter.loading_symbolcCr&)z'Returns the text colored like an error.Nr"r9r"r"r#errorr*z Printer.errorrLcCr&)z9Returns the string for a named emoji, or an empty string.Nr"r(rLr"r"r#emojir*z Printer.emojicCr&)z)Returns the text styled like a file path.Nr"r9r"r"r#filesr*z Printer.filesrowslist[list[str]]titlecCr&)z1Returns a grid of strings with an optional title.Nr")r(rWrYr"r"r#gridr*z Printer.gridcolumns list[str]cCr&)z2Returns the column text combined in a compact way.Nr"r(r[r"r"r#panel r*z Printer.panelrr%r-r.r,r/rr0Nr-r3r4r5rr0r-r8rr0)r;r/rr<rrD)rGrHrr8r-r3rr3rMr3r-r8rr3rPr<rr3rLr3rr3rWrXrYr8rr3r[r\rr3)__name__ __module__ __qualname____doc__ contextlibcontextmanagerabcabstractmethodr)r2r7r: staticmethodrCpropertyrErFrJrKrLrMrOrRrSrUrVrZr^r"r"r"r#rrsT           c@s eZdZdZejd ddZdS) DynamicTextz5A handle to a block of text that's allowed to change.r-r3rr0cCr&)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. Nr"r9r"r"r#set_textr*zDynamicText.set_textNr-r3rr0)rkrlrmrnrqrrrvr"r"r"r#rusrucs*eZdZd?fdd Zeejd@dd Zed d dAddZe dBdCddZ edBdDddZ edBdEddZ e edFddZe edFd d!ZedGd"d#ZedGd$d%ZedBdHd'd(ZedId*d+ZedGd,d-ZedJd0d1ZedGd2d3ZedGd4d5ZedBdKd9d:ZedLd=d>ZZS)Mr rrrr0cs$t||_tgd|_dS)N)-\|/)super__init__ _settings itertoolscycle _progress)r(r __class__r"r#r}!s z_PrinterTerm.__init__r%ccsn|jr |jjr dVdSt}|sdVn t|VWddSWddS1s0wYdSra)r~silentrr)_DynamicTermTextr(handler"r"r#r)&s  "z_PrinterTerm.dynamic_textNr+r-r.r,r/cCs@|jr |jjr dSt|ttfrd|n|}|||dSN )r~rr=listtuplejoin_display_fn_mappingr1r"r"r#r23sz_PrinterTerm.displayCallable[[str], None]cCst|}|tkr tjSt|krtkrtjSt|kr%tkr&tjSt|kr2tkr3tj St |kr?tkrCtj Stj Stj Sra) rrCrwandb termerrorrrtermwarnrtermlogrr+r"r"r#r@s  z _PrinterTerm._display_fn_mappingr3r4r5cCs4|jr |jjr dStjt|jd|dddS)N F)newline)r~rrrnextrr6r"r"r#r7Qs"z_PrinterTerm.progress_updater8cCs(|jr |jjr dS|p d}t|dS)NzO )r~rrrr9r"r"r#r:Xsz_PrinterTerm.progress_closerDcCr&)NFr"r'r"r"r#rE`z_PrinterTerm.supports_htmlcCstjtjSra)rutilis_unicode_safesysstderrr'r"r"r#rFesz_PrinterTerm.supports_unicodecCtj|dd}|S)NT)boldclickstyler(r-retr"r"r#rKjz_PrinterTerm.codecCr)Nyellowfgrrr"r"r#rLorz_PrinterTerm.namerMcCstj|ddd}|S)NblueT)r underliner)r(rMr-rr"r"r#rMtsz_PrinterTerm.linkrLcCs@t}tdkrtjtjrtddddddd}||d S) NWindowsu⭐️u🧹u🚀u🦍u🐢u️⚡)starbroomrocketgorillaturtle lightning) dictplatformsystemrrrrstdoutget)r(rLemojisr"r"r#rU{s z_PrinterTerm.emojicCtj|ddS)Nwhiterrr9r"r"r#rOsz_PrinterTerm.secondary_textrPr<cCs(|jsdS|tt}tjt|tdS)Nrr)rFlen_PROGRESS_SYMBOL_ANIMATIONrr_PROGRESS_SYMBOL_COLOR)r(rPidxr"r"r#rRs z_PrinterTerm.loading_symbolcCr)Nredrrr9r"r"r#rSz_PrinterTerm.errorcCstj|ddd}|S)NmagentaT)rrrrr"r"r#rVsz_PrinterTerm.filesrWrXrYcshtdd|Ddddt|ddgdfd d |D}|r/|d|dS|dS) Ncss|] }t|dVqdS)rN)r.0rowr"r"r# sz$_PrinterTerm.grid..rz {:>{max_len}}z{}rrrcsg|] }j|diqS)max_lenformatr format_rowrr"r# sz%_PrinterTerm.grid..)maxrrr(rWrYrZr"rr#rZs  z_PrinterTerm.gridr[r\cCsdd|Sr)rr]r"r"r#r^rz_PrinterTerm.panelrrrr0r_r`ra)r,r/rrrbrcrdrerfrhrgrirj)rkrlrmr}rrorpr)r2rsrr7r:rtrErFrKrLrMrUrOrRrSrVrZr^ __classcell__r"r"rr#r sP         r c@s$eZdZd ddZed d d Zd S)rrterm.DynamicBlockrr0cCs ||_dSra)_handlerr"r"r#r}s z_DynamicTermText.__init__r-r3cCs|j|dSra)rrvr9r"r"r#rvsz_DynamicTermText.set_textN)rrrr0rwrkrlrmr}rrvr"r"r"r#rs rcseZdZd<fdd Zeejd=dd Zed d d>ddZe ed?ddZ e ed?ddZ ed@ddZ ed@ddZ edAdBddZedCd!d"Zed@d#d$ZedDd'd(Zed@d)d*Zed@d+d,Ze dAdEd/d0ZedAdFd1d2ZedAdGd6d7ZedHd:d;ZZS)Irrrrr0cs0t||_t|_ddlm}||_dSNr)r2) r|r}r~r jupyter_progress_barrIPythonr2_ipython_display)r(rr2rr"r#r}s    z_PrinterJupyter.__init__r%ccs`|jr |jjr dVdS|jj|jddd}|r+t|V||jddSdVdS)NrT) display_id)r~rrr2HTML_DynamicJupyterTextupdaterr"r"r#r)s   z_PrinterJupyter.dynamic_textNr+r-r.r,r/cCsT|jr |jjr dSt|ttfrd|n|}d|}|j|j |dSNz
) r~rr=rrr splitlinesrr2rr1r"r"r#r2s z_PrinterJupyter.displayrDcCr&NTr"r'r"r"r#rErz_PrinterJupyter.supports_htmlcCr&rr"r'r"r"r#rFrz _PrinterJupyter.supports_unicoder3cCs d|dS)Nr"r9r"r"r#rK z_PrinterJupyter.codecC d|dS)Nz r"r9r"r"r#rLrz_PrinterJupyter.namerMr8cCsd|d|p|dS)Nzzr"rNr"r"r#rMsz_PrinterJupyter.linkrLcCr&Nrr"rTr"r"r#rUz_PrinterJupyter.emojicCs|Srar"r9r"r"r#rOrz_PrinterJupyter.secondary_textrPr<cCr&rr"rQr"r"r#rRrz_PrinterJupyter.loading_symbolcCr)Nzrr"r9r"r"r#rSrz_PrinterJupyter.errorcCr)Nrzr"r9r"r"r#rV rz_PrinterJupyter.filesr4r5cCs6|jr|jjs |js dS|durd}|j||dS)Ng?)r~rrrr6r"r"r#r7s z_PrinterJupyter.progress_updatecCs|jr |jdSdSra)rcloser9r"r"r#r:sz_PrinterJupyter.progress_closerWrXrYcsddddt|ddgdfdd|D}d|d }|r+d |d |d St|d S) Nrzz {}rzcsg|]}j|qSr"rrrr"r#r%sz(_PrinterJupyter.grid..zz
z

z


z
)rr_JUPYTER_TABLE_STYLESrr"rr#rZ"s  z_PrinterJupyter.gridr[r\cCs$ddd|D}td|dS)NrcSsg|]}d|dqS)z

r")rcolr"r"r#r-sz)_PrinterJupyter.panel..z
r)r_JUPYTER_PANEL_STYLES)r(r[rr"r"r#r^+sz_PrinterJupyter.panelrr_r`rdrerarfrhrgrbrcrirj)rkrlrmr}rrorpr)r2rtrErFrKrLrMrUrOrRrSrVr7r:rZr^rr"r"rr#rsN          rc@s$eZdZd ddZed ddZd S) rrr0cCsddlm}|j|_||_dSr)rr2r_ipython_to_htmlr)r(rr2r"r"r#r}2s  z_DynamicJupyterText.__init__r-r3cCs$d|}|j||dSr)rrrrrr9r"r"r#rv8sz_DynamicJupyterText.set_textN)rr0rwrr"r"r"r#r1s rra)rrrr)-rn __future__rrqrorrrtypingrrrtyping_extensionsrr wandb.util wandb.errorsr wandb.sdkrrr r rrrrrrrr_level_to_namer>rrrrr$ABCrrur rrrr"r"r"r#sl        v