Ë
    M²j  ã                   óv   — d dl Zd dl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y)
é    N)ÚAnyÚCheckpointExceptionÚexcÚreturnc                 óž   — t        j                  | j                  «      }|D ]&  }t        |d«      sŒt        j                  |dd «       Œ( | |fS )NÚ_code)ÚtbÚ
extract_tbÚ__traceback__ÚhasattrÚobjectÚ__setattr__)r   ÚsummaryÚframes      ú_/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/distributed/checkpoint/api.pyÚ_wrap_exceptionr   
   sO   € Ü�m‰m˜C×-Ñ-Ó.€Gð
 ò 5ˆÜ�5˜'Õ"Ü×Ñ˜u g¨tÕ4ð5ð �ˆ>Ðó    Úobjc                 ó¦   — t        | t        «      syt        | «      dk7  ryt        | d   t        «      xr t        | d   t        j
                  «      S )NFé   r   é   )Ú
isinstanceÚtupleÚlenÚBaseExceptionr	   ÚStackSummary)r   s    r   Ú_is_wrapped_exceptionr      sB   € Ü�cœ5Ô!ØÜ
ˆ3ƒx�1‚}ØÜ�c˜!‘fœmÓ,ÒT´¸CÀ¹FÄBÇOÁOÓ1TÐTr   c                   ób   ‡ — 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efd„Zˆ xZS )r   zNException raised if failure was detected as part of a checkpoint load or save.ÚmsgÚfailuresc                 ó4   •— t         ‰| �  ||«       || _        y )N)Ú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           	      óX  — d| j                   j                  «       › d�}| j                   j                  «       D ]n  \  }}|\  }}|d|› d�z  }|�'|dj                  t	        j
                  |«      «      z  }|dj                  t	        j                  t        |«      |¬«      «      z  }Œp |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ª   € Ø*¨4¯>©>×+>Ñ+>Ó+@Ð*AÀÐDˆØ"Ÿn™n×2Ñ2Ó4ò 	K‰NˆD�(Ø!‰JˆC�ØÐ>¸t¸fÀCÐHÑHˆCØÐ Ø�r—w‘wœrŸ~™~¨eÓ4Ó5Ñ5�Ø�2—7‘7œ2×3Ñ3´D¸³IÀSÔIÓJÑJ‰Cð	Kð ˆ
r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r2   ÚdictÚintÚWRAPPED_EXCEPTIONr#   Úpropertyr    r6   Ú__classcell__)r&   s   @r   r   r      sV   ø„ ÙXð"˜Cð "¨4°Ð5FÐ0FÑ+Gõ "ð ð˜$˜sÐ$5Ð5Ñ6ò ó ðð˜÷ r   )Ú	tracebackr	   Útypingr   r   r   r   r=   Ú__all__r   Úboolr   r   © r   r   ú<module>rE      s_   ðÛ Ý ð ˜-¨¯©Ð8Ñ9Ð à Ð
!€ð	˜ð 	Ð+<ó 	ðU˜sð U tó Uô˜-õ r   