
    2g'                     N    d dl mZ d dlmZ ddlmZ ddgiZdgZ G d de      Z	y	)
    )partial)Callback   )tqdmzgithub.com/	casperdclTqdmCallbackc                   B     e Zd ZdZddef fd	Zd Zd Zd Zd Z	 xZ
S )r   z Dask callback for task progress.Nc                 R    t         |   ||       |rt        |fi |}|| _        y)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/python.lazyprojects.co.uk/rdoDiscordBot/venv/lib/python3.12/site-packages/tqdm/dask.pyr   zTqdmCallback.__init__   s0     	ug6 ;{;J$    c                 X    | j                  t        fddD                    | _        y )Nc              3   :   K   | ]  }t        |           y wN)len).0kstates     r   	<genexpr>z,TqdmCallback._start_state.<locals>.<genexpr>   s      .P#NaCaM#Ns   )readywaitingrunningfinished)total)r   sumpbar)r   _r   s     `r   _start_statezTqdmCallback._start_state   s-    OO# .P#N.P +PO Q	r   c                 8    | j                   j                          y r   )r#   updater   r$   __s      r   	_posttaskzTqdmCallback._posttask    s    		r   c                 8    | j                   j                          y r   )r#   closer(   s      r   _finishzTqdmCallback._finish#   s    		r   c                 T    t        | j                  dd      }|yddlm}  ||       y)z*Displays in the current cell in Notebooks.	containerNr   )display)getattrbarnotebookr0   )r   r/   r0   s      r   r0   zTqdmCallback.display&   s(    DHHk48	%	r   )__name__
__module____qualname____doc__	tqdm_autor   r%   r*   r-   r0   __classcell__)r   s   @r   r   r      s'    *!4I %Qr   N)
	functoolsr   dask.callbacksr   autor   r8   
__author____all__r    r   r   <module>r@      s.     # #k]+

!8 !r   