
    [Th                     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                 F    U [         R                  " U R                  5      4$ N)tb
extract_tb__traceback__)r   s    X/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/checkpoint/api.py_wrap_exceptionr   
   s    s00122    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 r   )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__r1   dictintWRAPPED_EXCEPTIONr   propertyr   r5   __static_attributes____classcell__)r"   s   @r   r   r      s[    X"C "45F0F+G " $s$556    r   )	tracebackr	   typingr   r   r   r   r>   __all__r   boolr   r    r   r   <module>rG      s_      -89  
!3 3+< 3Us Ut U- r   