
    3j                     z    S SK rS SKJr  \\\R                  4   rS/rS\S\4S jr	S\S\
4S jr " S	 S\5      rg)
    N)AnyCheckpointExceptionexcreturnc                     [         R                  " U R                  5      nU H-  n[        US5      (       d  M  [        R                  USS 5        M/     X4$ )N_code)tb
extract_tb__traceback__hasattrobject__setattr__)r   summaryframes      Z/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/distributed/checkpoint/api.py_wrap_exceptionr   
   sK    mmC--.G
 5'""ugt4  >    objc                     [        U [        5      (       d  g[        U 5      S:w  a  g[        U S   [        5      =(       a    [        U S   [        R
                  5      $ )NF   r      )
isinstancetuplelenBaseExceptionr	   StackSummary)r   s    r   _is_wrapped_exceptionr      sE    c5!!
3x1}c!fm,TCFBOO1TTr   c                   r   ^  \ rS rSrSrS\S\\\4   4U 4S jjr	\
S\\\4   4S j5       rS\4S jrS	rU =r$ )
r      zNException raised if failure was detected as part of a checkpoint load or save.msgfailuresc                 0   > [         TU ]  X5        X l        g )N)super__init__	_failures)selfr    r!   	__class__s      r   r$   CheckpointException.__init__!   s    '!r   r   c                     U R                   $ )zYReturn a dictionary mapping node ranks to their associated exceptions in case of failure.r%   )r&   s    r   r!   CheckpointException.failures%   s     ~~r   c           	      X   SU R                   R                  5        S3nU R                   R                  5        Hm  u  p#Uu  pEUSU S3-  nUb(  USR                  [        R
                  " U5      5      -  nUSR                  [        R                  " [        U5      US95      -  nMo     U$ )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__CheckpointException.__str__*   s    *4>>+>+>+@*AD"nn224ND!JC>tfCHHC rwwr~~e455277233DISIJJC 5 
r   r*   )__name__
__module____qualname____firstlineno____doc__r6   dictintWRAPPED_EXCEPTIONr$   propertyr!   r:   __static_attributes____classcell__)r'   s   @r   r   r      s[    X"C "45F0F+G " $s$556    r   )	tracebackr	   typingr   r   r   r   rC   __all__r   boolr   r    r   r   <module>rL      s_      -89  
!	 	+< 	Us Ut U- r   