o
    W+ i                     @   s.   d dl mZ G dd dZG dd deZdS )    )tqdmc                   @   s4   e Zd ZdedefddZdefddZdd	 Zd
S )ProgressCallbackfilename	file_sizec                 C   s   || _ || _d S N)r   r   selfr   r    r	   c/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/modelscope/hub/callback.py__init__   s   
zProgressCallback.__init__sizec                 C      d S r   r	   r   r   r	   r	   r
   update
      zProgressCallback.updatec                 C   r   r   r	   r   r	   r	   r
   end   r   zProgressCallback.endN)__name__
__module____qualname__strintr   r   r   r	   r	   r	   r
   r      s    r   c                       s<   e Zd Zdedef fddZdefddZdd	 Z  ZS )
TqdmCallbackr   r   c              	      s@   t  || tddd|dkr|nddd| j d dd| _d S )	NBTi   r      zDownloading [])unitZ
unit_scaleZunit_divisortotalinitialZdescZleave)superr   r   r   progressr   	__class__r	   r
   r      s   zTqdmCallback.__init__r   c                 C   s   | j | d S r   )r    r   r   r	   r	   r
   r      s   zTqdmCallback.updatec                 C   s   | j   d S r   )r    closer   r	   r	   r
   r   !   s   zTqdmCallback.end)	r   r   r   r   r   r   r   r   __classcell__r	   r	   r!   r
   r      s    r   N)Z	tqdm.autor   r   r   r	   r	   r	   r
   <module>   s    