L ivddlZddlmZeeej fZdgZdedefdZ dede fdZ Gd deZ y) N)AnyCheckpointExceptionexcreturncD|tj|jfSN)tb extract_tb __traceback__)rs f/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torch/distributed/checkpoint/api.py_wrap_exceptionr s s001 22objct|tsyt|dk7ryt|dtxrt|dtj S)NFr) isinstancetuplelen BaseExceptionr StackSummary)rs r _is_wrapped_exceptionrsB c5 ! 3x1} c!fm , TCFBOO1TTrcbeZdZdZdedeeefffd Ze deeeffdZ defdZ xZ S)rzNException raised if failure was detected as part of a checkpoint load or save.msgfailuresc4t|||||_yr)super__init__ _failures)selfrr __class__s r rzCheckpointException.__init__s h'!rrc|jS)zYReturn a dictionary mapping node ranks to their associated exceptions in case of failure.)r)r s r rzCheckpointException.failuress~~rc Xd|jjd}|jjD]n\}}|\}}|d|dz }|'|djt j |z }|djt j t||z }p|S)NzCheckpointException ranks: z)Traceback (most recent call last): (RANK z) )value)rkeysitemsjoinr format_listformat_exception_onlytype)r strrankexc_pairrtraces r __str__zCheckpointException.__str__"s*4>>+>+>+@*AD"nn224 KND(!JC >tfCH HC rwwr~~e455 277233DISIJ JC  K  r) __name__ __module__ __qualname____doc__r-dictintWRAPPED_EXCEPTIONrpropertyrr1 __classcell__)r!s@r rrsVX"C"45F0F+G"$s$556r) tracebackr typingrrrrr8__all__r boolrrrr r@s_-89 !33+<3UsUtU-r