
      i                        d Z ddlZddlZddlmZ ddlmc mZ ddlm	Z	m
Z
mZmZmZmZ  ej                  e      e_        dee   fdZej(                  Z eed e       xs d      e_         G d d	e      Zee_        y)
zFUtilities that can be used for calling functions on a particular rank.    N)Optional)WarningCacherank_prefixed_messagerank_zero_debugrank_zero_deprecationrank_zero_inforank_zero_warnreturnc                  r    d} | D ]0  }t         j                  j                  |      }|%t        |      c S  y )N)RANK
LOCAL_RANKSLURM_PROCIDJSM_NAMESPACE_RANK)osenvirongetint)	rank_keyskeyranks      y/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_fabric/utilities/rank_zero.py	_get_rankr   #   s;     MIzz~~c"t9 
     r   c                       e Zd ZdZy)LightningDeprecationWarningz)Deprecation warnings raised by Lightning.N)__name__
__module____qualname____doc__ r   r   r   r   5   s    3r   r   )r   loggingr   typingr   "lightning_utilities.core.rank_zerocore	rank_zerorank_zero_moduler   r   r   r   r   r	   	getLoggerr   logr   r   rank_zero_onlygetattrr   DeprecationWarningr   rank_zero_deprecation_categoryr    r   r   <module>r-      s    M  	  = =  )w((2  	8C= 	 "00 nfik6FQG 4"4 4 3N  /r   