�
���g� � � � U d Z ddlZddlmZ ddlmZ epdaee d< dd �Z
dd
�Z defd �Z G d � d
e� � Ze Z
e
Zd� ZdS )a� Utility helpers to handle progress bars in `datasets`.
Example:
1. Use `datasets.utils.tqdm` as you would use `tqdm.tqdm` or `tqdm.auto.tqdm`.
2. To disable progress bars, either use `disable_progress_bars()` helper or set the
environment variable `HF_DATASETS_DISABLE_PROGRESS_BARS` to 1.
3. To re-enable progress bars, use `enable_progress_bars()`.
4. To check whether progress bars are disabled, use `are_progress_bars_disabled()`.
NOTE: Environment variable `HF_DATASETS_DISABLE_PROGRESS_BARS` has the priority.
Example:
```py
from datasets.utils import (
are_progress_bars_disabled,
disable_progress_bars,
enable_progress_bars,
tqdm,
)
# Disable progress bars globally
disable_progress_bars()
# Use as normal `tqdm`
for _ in tqdm(range(5)):
do_something()
# Still not showing progress bars, as `disable=False` is overwritten to `True`.
for _ in tqdm(range(5), disable=False):
do_something()
are_progress_bars_disabled() # True
# Re-enable progress bars globally
enable_progress_bars()
# Progress bar will be shown !
for _ in tqdm(range(5)):
do_something()
```
� N)�tqdm� )�!HF_DATASETS_DISABLE_PROGRESS_BARSF�#_hf_datasets_progress_bars_disabled�returnc �H � t du rt j d� � dS dadS )z�
Disable globally progress bars used in `datasets` except if `HF_DATASETS_DISABLE_PROGRESS_BAR` environment
variable has been set.
Use [`~utils.enable_progress_bars`] to re-enable them.
FzpCannot disable progress bars: environment variable `HF_DATASETS_DISABLE_PROGRESS_BAR=0` is set and has priority.NT�r �warnings�warnr � � �c/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/datasets/utils/tqdm.py�disable_progress_barsr < s; � � )�E�1�1��
�
�
�
�
� ��*.�'�'�'r
c �H � t du rt j d� � dS dadS )z�
Enable globally progress bars used in `datasets` except if `HF_DATASETS_DISABLE_PROGRESS_BAR` environment
variable has been set.
Use [`~utils.disable_progress_bars`] to disable them.
TzoCannot enable progress bars: environment variable `HF_DATASETS_DISABLE_PROGRESS_BAR=1` is set and has priority.NFr r r
r �enable_progress_barsr M s; � � )�D�0�0��
�
�
�
�
� ��*/�'�'�'r
c � � t S )a! Return whether progress bars are globally disabled or not.
Progress bars used in `datasets` can be enable or disabled globally using [`~utils.enable_progress_bars`]
and [`~utils.disable_progress_bars`] or by setting `HF_DATASETS_DISABLE_PROGRESS_BAR` as environment variable.
)r r r
r �are_progress_bars_disabledr ^ s
� � /�.r
c �6 � � e Zd ZdZ� fd�Zdeddf� fd� Z� xZS )r z�
Class to override `disable` argument in case progress bars are globally disabled.
Taken from https://github.com/tqdm/tqdm/issues/619#issuecomment-619639324.
c �` �� t � � rd|d< t � � j |i |�� d S )NT�disable)r �super�__init__)�self�args�kwargs� __class__s �r r z
tqdm.__init__o s= �� � %� '� '� %� $�F�9� ������$�)�&�)�)�)�)�)r
�attrr Nc �| �� t � � � |� � dS # t $ r |dk r� Y dS w xY w)z;Fix for https://github.com/huggingface/datasets/issues/6066�_lockN)r � __delattr__�AttributeError)r r r s �r r ztqdm.__delattr__t sV �� � � �G�G� � �� %� %� %� %� %�� � � � ��w���� ��� ���s �!& �;�;)�__name__�
__module__� __qualname__�__doc__r �strr �
__classcell__)r s @r r r h sn �� � � � � �� � *� *� *� *� *�
�� �� � � � � � � � � � r
r c � � t � � S )N)r r r
r �is_progress_bar_enabledr) � s � �)�+�+� +�+r
)r N)r% r
� tqdm.autor �old_tqdm�configr r �bool�__annotations__r r r �enable_progress_bar�disable_progress_barr) r r
r �<module>r1 s� ��(� (� (�T ���� &� &� &� &� &� &� 6� 6� 6� 6� 6� 6� -N�,V�QV� #�T� V� V� V�/� /� /� /�"0� 0� 0� 0�"/�D� /� /� /� /�� � � � �8� � � �, +� �,� �,� ,� ,� ,� ,r