
    IЦi                         S SK rS SKJrJr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DictTupleCheckpointExceptionexcreturnc                 F    U [         R                  " U R                  5      4$ N)tb
extract_tb__traceback__)r   s    _/var/www/html/ai-image-ml/venv/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                   j   ^  \ 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 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__r3   r   intWRAPPED_EXCEPTIONr!   propertyr   r7   __static_attributes____classcell__)r$   s   @r   r   r      sT    X"C "45F0F+G " $s$556   r   )	tracebackr   typingr   r   r   r   r   r?   __all__r   boolr   r    r   r   <module>rH      sb     # # -89  
!3 3+< 3Us Ut U- r   