o
    oi/                     @   s   d dl Z d dlZd dlmZmZ d dlmZmZ d dlm	Z
 dedefddZeed	ee jd
d e_dededdfddZdededdfddZdededdfddZeeZeeZeeZeejedZdS )    N)partialwraps)AnyCallable)_loggerfnreturnc                    s&   t  dtdtdtf fdd}|S )Nargskwargsr   c                     s   t jdkr | i |S d S )Nr   )rank_zero_onlyrankr	   r
   r    [/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torchmetrics/utilities/prints.py
wrapped_fn   s   
z"rank_zero_only.<locals>.wrapped_fn)r   r   )r   r   r   r   r   r      s   r   r   
LOCAL_RANKr	   r
   c                  O      t j| i | d S N)warningswarnr   r   r   r   _warn#      r   c                  O   r   r   )loginfor   r   r   r   _info'   r   r   c                  O   r   r   )r   debugr   r   r   r   _debug+   r   r   )category)osr   	functoolsr   r   typingr   r   torchmetricsr   r   r   getattrintenvirongetr   r   r   r   rank_zero_debugrank_zero_inforank_zero_warnr   FutureWarning_future_warningr   r   r   r   <module>   s   
