o
Ci* @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ dZ
zddl Z dZ
W n? e yf d Z
ddl
Z
e
$ e
jd
d d z ddlm mZ W n e yT Y nw W d n1 s_w Y Y nw z-e
d krdd
lmZ ddlmZ ddlmZ dZ
ndd
l mZ ddl mZ ddl mZ W n/ e y zdd
lmZ ddlmZ ddlmZ dZ
W n e y dZ
dZeZY nw Y nw zddlmZ W n e y Y nw dg diZg dZ dZ G dd deZ G dd d e Z d d Z e Ze Z!dS ) z
IPython/Jupyter Notebook progressbar decorator for iterators.
Includes a default `range` iterator printing to `stderr`.
Usage:
>>> from tqdm.notebook import trange, tqdm
>>> for i in trange(10):
... ...
N)escape)proxy )tqdmT ignorez2.*The `IPython.html` package has been deprecated.*)message)HTML)
FloatProgress)HBox )ContainerWidget)FloatProgressWidget )displayz github.com/)lrq3000 casperdcl alexanderkuk)
tqdm_notebooktnranger trangez|IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.htmlc s4 e Zd ZdZd
ddZd fdd Zdd Z ZS ) TqdmHBoxz.`ipywidgets.HBox` with a pretty representationNc C s4 t | dd }|d u r i S |j}|d ur| |d< |S )Npbarascii)getattr format_dict)selfprettyr d r e/var/www/html/fyndo/python/python_agents/rag_suite/venv/lib/python3.10/site-packages/tqdm/notebook.py_json_G s
zTqdmHBox._json_Fc s8 t | dd }|d u rtt| S |jdi | |S )Nr r )r superr __repr__ format_meterr" )r r r __class__r r! r$ P s zTqdmHBox.__repr__c O s | | d d S )NT)textr$ )r pp___r r r!
_repr_pretty_V s zTqdmHBox._repr_pretty_N)F)__name__
__module__ __qualname____doc__r" r$ r,
__classcell__r r r& r! r E s
r c s e Zd ZdZedddZ dddZed d
Zej d d
Z fd d
Z
fddZ d fdd Z fddZ
dd Zd fdd Z ZS ) r zB
Experimental IPython/Jupyter Notebook widget using tqdm!
Nc C s t du rtt|rt d|d}nt ddd}d|_d|_|du r%d|j_t }t }|r0||_t|||gd}|durft |}z t
|dkrK|d7 }W n t yU Y nw d |j_ ||j_d
|j_
d |j_|S ) zY
Manage the printing of an IPython/Jupyter Notebook progress bar widget.
Nr )minmaxr info20px)childrenpx2z inline-flexzrow wrap) IProgress ImportError WARN_NOIPYWvalue bar_stylelayoutwidthr
r strint
ValueErrorflexr flex_flow)r* totaldescncolsr ltextrtext containerr r r! status_printer^ s8
z tqdm_notebook.status_printerFTc C s$ |s|s| j }|d p ddd|d< | jdi |}| jj\}}} | j|_|rId|v r:ttt j
d|dd\}
} ndt|}
} |
|_| rI| | _|rW|j d ksT|d
krW||_ |rx|j d krxz| j W n
t
yr d | j_Y nw d | jj_|r| jd
kr| jst| j d| _d S d S d S d S )N
bar_formatz{l_bar}{r_bar}z{bar}zz \|?\|?r )maxsplit dangersuccessFhiddenr Tr )r replacer% rK r7 nr= mapr resplitr> closeAttributeErrorvisibler?
visibilitydelay displayedr ) r msgposrX r> check_delayr rI r rJ leftrightr r r! r s:
ztqdm_notebook.displayc C s t | dr
| jjd jjS d S NrK hasattrrK r7 style bar_colorr r r r! colour s
ztqdm_notebook.colourc C s" t | dr|| jjd j_d S d S rc re )r rh r r r! rj s
c sN | }|dtj}|tju s|du rtj|d< d|d< t|dd|d< |dd}|dd}tt| j |i | | j
sE|d sLd d
| _ dS | j rQd n|d d| _
| jdu r_d
n| jpcd
}| jrl| j| n| j}| | j|| j| j
| _t| | j_d| _|r| jdkrt| j d| _| j| _ || _| j
s| jdd dS dS )z
Supports the usual `tqdm.tqdm` parameters as well as those listed below.
Parameters
----------
display : Whether to call `display(self.container)` immediately
[default: True].
fileNTguidisableFrj r c _ d S r- r )r* r+ r r r! s z(tqdm_notebook.__init__..z100%rH r r )r` )copygetsysstderrstdoutboolpopr# r __init__rm disp
dynamic_ncolsrH
unit_scalerF rL fprG rK r r r] r\ r rj )r argskwargs
file_kwargrj display_hererz rF r& r r! rw s4
ztqdm_notebook.__init__c # s< zt t| }|D ]}|V q W d S | jdd )NrP r> )r# r __iter__rx )r itobjr& r r! r s ztqdm_notebook.__iter__r c s* z
t t| j|dW S | jdd )N)rT rP r )r# r updaterx )r rT r& r r! r s
ztqdm_notebook.updatec sf | j rd S tt| | jr | j| jk r | jddd d S | jr*| jddd d S | jddd d S )NrP F)r> r` rQ T)rX r` )rm r# r rX rF rT rx leaveri r& r r! rX s ztqdm_notebook.closec O rn r- r )r r* r+ r r r! clear s ztqdm_notebook.clearc sd | j r tt| j|dS | jj\}}}d|_|dur)||_| js)| j du r)d|j
_ tt| j|dS )z
Resets to 0 iterations for repeated use.
Consider combining with `leave=True`.
Parameters
----------
total : int or float, optional. Total to use for the new bar.
)rF rO N) rm r# r resetrK r7 r> r4 rF rH r? r@ )r rF r* r r& r r! r s
ztqdm_notebook.reset)NNN)NNFNT)r r- )r. r/ r0 r1 staticmethodrL r propertyrj setterrw r r rX r r r2 r r r& r! r Z s" ,
4
-
r c O s t t| fi |S )z:Shortcut for `tqdm.notebook.tqdm(range(*args), **kwargs)`.)r range)r| r} r r r! r 5 s r )"r1 rV rr htmlr weakrefr stdr std_tqdmIPY
ipywidgetsr; warningscatch_warningsfilterwarningsIPython.html.widgetswidgetsr
r r: r r r objectIPython.displayr
__author____all__r< r r r r r r r r! s
\