
    h'                     N    S SK Jr  S SKJr  SSKJr  SS/0rS/r " S S\5      r	g	)
    )partial)Callback   )tqdmzgithub.com/	casperdclTqdmCallbackc                   N   ^  \ rS rSrSrSS\4U 4S jjrS rS rS r	S r
S	rU =r$ )
r      z Dask callback for task progress.Nc                 R   > [         TU ]  XS9  U(       a  [        U40 UD6nX0l        g)z
Parameters
----------
tqdm_class  : optional
    `tqdm` class to use for bars [default: `tqdm.auto.tqdm`].
tqdm_kwargs  : optional
    Any other arguments used for all bars.
)startpretaskN)super__init__r   
tqdm_class)selfr   r   r   tqdm_kwargs	__class__s        ;/var/www/html/env/lib/python3.13/site-packages/tqdm/dask.pyr   TqdmCallback.__init__   s,     	u6 ;{;J$    c                 T   ^ U R                  [        U4S jS 5       5      S9U l        g )Nc              3   @   >#    U  H  n[        TU   5      v   M     g 7fN)len).0kstates     r   	<genexpr>,TqdmCallback._start_state.<locals>.<genexpr>   s       .P#NaCaMM#Ns   )readywaitingrunningfinished)total)r   sumpbar)r   _r   s     `r   _start_stateTqdmCallback._start_state   s-    OO# .P#N.P +PO Q	r   c                 8    U R                   R                  5         g r   )r&   updater   r'   __s      r   	_posttaskTqdmCallback._posttask    s    		r   c                 8    U R                   R                  5         g r   )r&   closer,   s      r   _finishTqdmCallback._finish#   s    		r   c                 V    [        U R                  SS5      nUc  gSSKJn  U" U5        g)z*Displays in the current cell in Notebooks.	containerNr   )display)getattrbarnotebookr6   )r   r5   r6   s      r   r6   TqdmCallback.display&   s(    DHHk48	%	r   )r&   r   )__name__
__module____qualname____firstlineno____doc__	tqdm_autor   r(   r.   r2   r6   __static_attributes____classcell__)r   s   @r   r   r      s,    *!4I %Q r   N)
	functoolsr   dask.callbacksr   autor   r@   
__author____all__r    r   r   <module>rI      s.     # #k]+

!8 !r   