
    a,j                     z    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dS )
    N)AnyCheckpointExceptionexcreturnc                     t          j        | j                  }|D ].}t          |d          rt                              |dd            /| |fS )N_code)tb
extract_tb__traceback__hasattrobject__setattr__)r   summaryframes      `/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/torch/distributed/checkpoint/api.py_wrap_exceptionr   
   sZ    mC-..G
  5 55'"" 	5ugt444>    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    c5!! u
3xx1}}uc!fm,,TCFBO1T1TTr   c                   t     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                 Z    t                                          ||           || _        d S )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           	      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    D4>+>+>+@+@DDD"n2244 	K 	KND(!JCHtHHHHC rwwr~e4455527723DIISIIIJJJCC
r   )__name__
__module____qualname____doc__r2   dictintWRAPPED_EXCEPTIONr#   propertyr    r6   __classcell__)r&   s   @r   r   r      s        XX"C "45F0F+G " " " " " " $s$556    X        r   )	tracebackr	   typingr   r   r   r   r=   __all__r   boolr   r    r   r   <module>rE      s              -89  
!	 	+< 	 	 	 	Us Ut U U U U    -     r   