+
    –°³j3  ã                   ó"  a € R t0 t ^ RIt^ RIt^ RIt^ RIHt ^ RIHtHt ^ RI	H
t
 ^ RIHt ^ RIHu Ht ^ RIHt ]P$                  ! 4       t. t] ^ k ]P*                  ! ]4      s]! R4      t]
! R4      tR	 R
 ltR R ltRR R lltR R ltR# )é    N)ÚCallable)ÚAnyÚTypeVar)Ú	ParamSpec)Úuuid4)ÚDCP_LOGGER_NAMEÚ_TÚ_Pc                óF   € V ^8„  d   QhR\         \        \        3,          /# ©é   Úreturn©ÚdictÚstrr   )Úformats   "Úd/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/checkpoint/logger.pyÚ__annotate__r      s   € ÷  ñ  ´t¼CÄ¸Hµ~ñ  ó    c                 ó  € / pVP                  R4      pVP                  R4      pVP                  R4      pVP                  R4      pV'       g"   T;'       g    T;p'       d   \        VRR4      pVe   \        V4      MTVR&   \        \        4       P                  4      VR&   V'       d   VP
                  P                  VR&   V'       d   VP
                  P                  VR&   V'       d   VP
                  P                  VR&   V# )z(
Extracts log data from dcp method args
Ústorage_writerÚstorage_readerÚplannerÚcheckpoint_idNÚuuid)ÚgetÚgetattrr   r   ÚintÚ	__class__Ú__name__)ÚargsÚkwargsÚmsg_dictr   r   r   r   Ú
serializers   *,      r   Ú_msg_dict_from_dcp_method_argsr%      sì   € ð €Hð —Z‘ZÐ 0Ó1€NØ—Z‘ZÐ 0Ó1€NØ�j‰j˜Ó#€Gà—J‘J˜Ó/€Mß¨N×,LÐ,L¸nÐL˜jÖLÜ 
¨O¸TÓBˆð ,Ò7ŒˆMÔ¸]ð ˆ_Ñô œ5›7Ÿ;™;Ó'€HˆVÑçØ%3×%=Ñ%=×%FÑ%FˆÐ!Ñ"çØ%3×%=Ñ%=×%FÑ%FˆÐ!Ñ"çØ%×/Ñ/×8Ñ8ˆ�Ñà€Or   c                óF   € V ^8„  d   QhR\         \        \        3,          /# r   r   )r   s   "r   r   r   >   s   € ÷ ñ ´´c¼3°hµñ r   c                 ón   € \        V/ VB pVP                  \        P                  ! V .VO5/ VB 4       V# )N)r%   ÚupdateÚc10d_loggerÚ_get_msg_dict)Ú	func_namer!   r"   r#   s   &*, r   r*   r*   >   s7   € Ü-¨tÐ>°vÑ>€HØ‡O�O”K×-Ò-¨iÐI¸$ÒIÀ&ÑIÔJà€Or   c          
      ó¤   € V ^8„  d   QhR\         R\        R\        \        \        \        3,          .\        \        \        3,          3,          /# )r   Úlog_exceptionsÚwrapper_kwargsr   )Úboolr   r   r
   r	   )r   s   "r   r   r   E   sC   € ÷ )ñ )Üð)Ü47ð)äŒxœœB˜ÕÐ ¤(¬2¬r¨6Õ"2Ð2Õ3ñ)r   c                ó   a a€ R V V3R llpV# )zKThis method decorator logs the start, end, and exception of wrapped events.c                óF   € V ^8„  d   QhR\         \        \        3,          /# )r   Úfunc)r   r
   r	   )r   s   "r   r   Ú(_dcp_method_logger.<locals>.__annotate__J   s   € ÷ "ñ "œ¤¤R Õ(ñ "r   c                 óR   <a € \         P                  ! S 4      R  V VV3R ll4       pV# )c                ód   € V ^8„  d   QhR\         P                  R\         P                  R\        /# )r   r!   r"   r   )r
   r!   r"   r	   )r   s   "r   r   Ú;_dcp_method_logger.<locals>.decorator.<locals>.__annotate__L   s)   € ÷ 	ñ 	œ2Ÿ7™7ð 	¬b¯i©ið 	¼Bñ 	r   c                  ó  <€ \        SP                  .V O5/ / S	CVCB pR VR&   \        P                  ! 4       pW2R&   SVR&   \        P                  V4        S! V / VB pRTR&   \        P                  ! 4       p\        P                  ! 4       TR&   Yc,
          TR&   \        P                  T4       T#   \         dG   pS'       d9   RTR&   T TR&   \        P                  ! 4       TR&   \        P                  T4       h Rp?ii ; i)	ÚstartÚeventÚtimer-   Ú	exceptionÚerrorNÚendÚtimes_spent)r*   r    r:   Útime_nsÚ_dcp_loggerÚdebugÚBaseExceptionr<   )
r!   r"   r#   Út0Úresultr<   Út1r2   r-   r.   s
   *,     €€€r   ÚwrapperÚ6_dcp_method_logger.<locals>.decorator.<locals>.wrapperK   s  ø€ ä$Ø—‘ðØ $òØ(D¨>Ð(D¸VÐ(DñˆHð
 !(ˆH�WÑÜ—’“ˆBØ!�VÑØ)7ˆHÐ%Ñ&Ü×Ñ˜hÔ'ðÙ˜tÐ. vÑ.�ð !&ˆH�WÑÜ—’“ˆBÜ#Ÿ|š|›~ˆH�VÑØ&(¥gˆH�]Ñ#Ü×Ñ˜hÔ'àˆMøô !ô ß!Ø(3�H˜WÑ%Ø+0¨'�H˜WÑ%Ü'+§|¢|£~�H˜VÑ$Ü×%Ñ% hÔ/Øûðús   ÁB4 Â4DÂ?AD Ä D)Ú	functoolsÚwraps)r2   rF   r-   r.   s   f €€r   Ú	decoratorÚ%_dcp_method_logger.<locals>.decoratorJ   s+   ù€ Ü	�Š˜Ó	÷	ñ 	ó 
ð	ð@ ˆr   © )r-   r.   rJ   s   fl r   Ú_dcp_method_loggerrM   E   s   ù€ ÷
"ñ "ðH Ðr   c                ó$   € V ^8„  d   QhR\         /# )r   Úrank)r   )r   s   "r   r   r   q   s   € ÷ ñ ”sñ r   c                 ó4  € \         P                  \        P                  4       \        P                  ! 4       pVP                  \        P                  4       \        P
                  ! R V  R24      pVP                  V4       \         P                  V4       R# )Ú[z6] %(asctime)s - %(name)s - %(levelname)s - %(message)sN)ÚloggerÚsetLevelÚloggingÚINFOÚStreamHandlerÚ	FormatterÚsetFormatterÚ
addHandler)rO   ÚchÚ	formatters   &  r   Ú_init_loggerr\   q   si   € Ü
‡O�O”G—L‘LÔ!Ü	×	Ò	Ó	 €BØ‡K�K”—‘ÔÜ×!Ò!Ø
ˆDˆ6ÐGÐHó€Ið ‡O�O�IÔÜ
×Ñ�bÖr   c                óV   € V ^8„  d   Qh/ ^ \         9   d   \        \        ,          ;R&   # )r   Ú__all__)Ú__conditional_annotations__Úlistr   )r   s   "r   r   r      s   € × Ñ ÷ Ò ŒŒc�Ñ ò r   )F)r_   rH   rT   r:   Úcollections.abcr   Útypingr   r   Útyping_extensionsr   r   r   Útorch.distributed.c10d_loggerÚdistributedr)   Ú-torch.distributed.checkpoint.logging_handlersr   Ú	getLoggerrR   r^   Ú_get_or_create_loggerr@   r	   r
   r%   r*   rM   r\   r   )r_   s   @r   Ú<module>ri      s{   øðç Ð Û Û Ý $ß Ý 'Ý ç 3Ð 3Ý Ið 
×	Ò	Ó	€ð €Ó ð ×/Ò/°Ó@€áˆTƒ]€Ùˆtƒ_€õ õF÷)÷Xr   