dask.cpython-312.pyc•2.56 kB
�
lMg' � �N � d dl mZ d dlmZ ddlmZ ddgiZdgZ G d� de� Z y )
� )�partial)�Callback� )�tqdmzgithub.com/� casperdcl�TqdmCallbackc �B � � e Zd ZdZddef� fd� Zd� Zd� Zd� Zd� Z � xZ
S )r z Dask callback for task progress.Nc �R �� t �| � ||�� |rt |fi |��}|| _ 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 ��UC:\Users\noahv\Documents\GitHub\clickup-operator\.venv\Lib\site-packages\tqdm/dask.pyr zTqdmCallback.__init__
s0 �� � ���u�g��6�� ��;�{�;�J�$��� c �X �� | j t �fd�dD � � �� | _ y )Nc 3 �: �K � | ] }t �| � �� � y �w�N)�len)�.0�k�states �r � <genexpr>z,TqdmCallback._start_state.<locals>.<genexpr> s �� �� � .P�#N�a�C��a��M�#N�s �)�ready�waiting�running�finished)�total)r �sum�pbar)r �_r s `r �_start_statezTqdmCallback._start_state s- �� ��O�O�#� .P�#N�.P� +P�O� Q�� r c �8 � | j j � y r )r# �update�r r$ �__s r � _posttaskzTqdmCallback._posttask s � �� � ���r c �8 � | j j � y r )r# �closer( s r �_finishzTqdmCallback._finish# s � �� � ���r c �T � t | j dd� }|�yddlm} ||� y)z*Displays in the current cell in Notebooks.� containerNr )�display)�getattr�bar�notebookr0 )r r/ r0 s r r0 zTqdmCallback.display&