§
    Ÿ²ÅjÕ  ã                   ó‚   — d dl Zd dlmZmZmZ eeej        f         ZdgZ	dedefd„Z
dedefd„Z G d	„ de¦  «        ZdS )
é    N)ÚAnyÚDictÚTupleÚCheckpointExceptionÚexcÚreturnc                 ó8   — | t          j        | j        ¦  «        fS ©N)ÚtbÚ
extract_tbÚ__traceback__)r   s    úQC:\Projects\Historical\venv\Lib\site-packages\torch/distributed/checkpoint/api.pyÚ_wrap_exceptionr      s   € Ø•”˜sÔ0Ñ1Ô1Ð2Ð2ó    Úobjc                 óÐ   — t          | t          ¦  «        sdS t          | ¦  «        dk    rdS t          | d         t          ¦  «        ot          | d         t          j        ¦  «        S )NFé   r   é   )Ú
isinstanceÚtupleÚlenÚBaseExceptionr   ÚStackSummary)r   s    r   Ú_is_wrapped_exceptionr      sV   € Ý�c�5Ñ!Ô!ð ØˆuÝ
ˆ3�x„x�1‚}€}ØˆuÝ�c˜!”f�mÑ,Ô,ÐTµ¸CÀ¼FÅBÄOÑ1TÔ1TÐTr   c                   ón   ‡ — e Zd ZdZdedeeef         fˆ fd„Ze	deeef         fd„¦   «         Z
d„ Zˆ xZS )r   zNException raised if failure was detected as part of a checkpoint load or save.ÚmsgÚfailuresc                 óZ   •— t          ¦   «                              ||¦  «         || _        d S r
   )ÚsuperÚ__init__Ú	_failures)Úselfr   r   Ú	__class__s      €r   r    zCheckpointException.__init__   s(   ø€ Ý‰Œ×Ò˜˜hÑ'Ô'Ð'Ø!ˆŒˆˆr   r   c                 ó   — | j         S )zYReturn a dictionary mapping node ranks to their associated exceptions in case of failure.)r!   )r"   s    r   r   zCheckpointException.failures   s   € ð Œ~Ðr   c           	      ód  — d| j                              ¦   «         › d�}| j                              ¦   «         D ]x\  }}|\  }}|d|› d�z  }|�*|d                     t	          j        |¦  «        ¦  «        z  }|d                     t	          j        t          |¦  «        |¬¦  «        ¦  «        z  }Œy|S )NzCheckpointException ranks:ú
z)Traceback (most recent call last): (RANK z)
Ú )Úvalue)r!   ÚkeysÚitemsÚjoinr   Úformat_listÚformat_exception_onlyÚtype)r"   ÚstrÚrankÚexc_pairr   Útraces         r   Ú__str__zCheckpointException.__str__#   s»   € ØD¨4¬>×+>Ò+>Ñ+@Ô+@ÐDÐDÐDˆØ"œn×2Ò2Ñ4Ô4ð 	Kð 	K‰NˆD�(Ø!‰JˆC�ØÐH¸tÐHÐHÐHÑHˆCØÐ Ø�r—w’w�rœ~¨eÑ4Ô4Ñ5Ô5Ñ5�Ø�2—7’7�2Ô3µD¸±I´IÀSÐIÑIÔIÑJÔJÑJˆCˆCØˆ
r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r/   r   ÚintÚWRAPPED_EXCEPTIONr    Úpropertyr   r3   Ú__classcell__)r#   s   @r   r   r      s�   ø€ € € € € ØXÐXð"˜Cð "¨4°Ð5FÐ0FÔ+Gð "ð "ð "ð "ð "ð "ð ð˜$˜sÐ$5Ð5Ô6ð ð ð ñ „Xððð ð ð ð ð ð r   )Ú	tracebackr   Útypingr   r   r   r   r   r9   Ú__all__r   Úboolr   r   © r   r   ú<module>rA      sÊ   ðà Ð Ð Ð Ø #Ð #Ð #Ð #Ð #Ð #Ð #Ð #Ð #Ð #ð ˜-¨¬Ð8Ô9Ð à Ð
!€ð3˜ð 3Ð+<ð 3ð 3ð 3ð 3ðU˜sð U tð Uð Uð Uð Uðð ð ð ð ˜-ñ ô ð ð ð r   