+
    –°³já  ã                   ój   € ^ RI Ht ^RIHt ^RIHtHtHtHtH	t	  ]
]]3,          tR R ltR R ltR# )	é    )ÚSTATE_DICT_TYPE)Ú_version)ÚOBJ_PATHÚset_elementÚSTATE_DICT_ITEMÚtraverse_state_dictÚtraverse_state_dict_v_2_3c                óR   € V ^8„  d   QhR\         R\        \         \        3,          /# )é   Ú
state_dictÚreturn)r   ÚtupleÚFLATTEN_MAPPING)Úformats   "Új/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/checkpoint/_nested_dict.pyÚ__annotate__r      s&   € ÷  ñ  Üð ä
Œ?œOÐ+Õ,ñ ó    c                óÈ   aa€ / o/ oR VV3R llp\         P                  RJ;'       d    \         P                  R8H  pV'       d   \        W4       SS3# \        W4       SS3# )ao  
Flatten ``state_dict`` made of nested dicts and lists into a top level dictionary.

Use ``unflatten_state_dict`` to revert this process.
Returns:
    A tuple with the flatten state_dict and a mapping from original to new state_dict.
N.B. The new keys are derived from the object paths, joined by dot.
    For example: ``{ 'a': {'b':...}}`` results in the key `a.b`.
c                ó4   € V ^8„  d   QhR\         R\        RR/# )r   ÚpathÚvaluer   N)r   r   )r   s   "r   r   Ú(flatten_state_dict.<locals>.__annotate__*   s!   € ÷ !ñ !œð !¬ð !¸Tñ !r   c                 ó„   <€ R P                  \        \        V 4      4      pVS9   d   \        RV 24      hVSV&   V SV&   R# )Ú.zduplicated flatten key N)ÚjoinÚmapÚstrÚ
ValueError)r   r   Únew_fqnÚ	flattenedÚmappingss   && €€r   Ú	flat_copyÚ%flatten_state_dict.<locals>.flat_copy*   sF   ø€ Ø—(‘(œ3œs D›>Ó*ˆØ�iÔÜÐ6°w°iÐ@ÓAÐAØ"ˆ	�'ÑØ ˆ�Ór   NÚ2_3)r   Ú_derived_versionr	   r   )r   r"   Ú	use_v_2_3r    r!   s   &  @@r   Úflatten_state_dictr'      sp   ù€ ð "$€IØ "€H÷!ñ !ô 	×!Ñ!¨Ð-×TÐT´(×2KÑ2KÈuÑ2Tð ÷ Ü! *Ô8ð �hÐÐô 	˜JÔ2Ø�hÐÐr   c                ó<   € V ^8„  d   QhR\         R\        R\         /# )r   r   Úmappingr   )r   r   )r   s   "r   r   r   >   s$   € ÷ ñ ÜðÜ*9ðäñr   c                ób   € / pV P                  4        F  w  r4\        W!V,          V4       K  	  V# )zaRestore the original nested state_dict according to ``mapping`` and the flattened ``state_dict``.)Úitemsr   )r   r)   ÚnestedÚkeyr   s   &&   r   Úunflatten_state_dictr.   >   s1   € ð !€FØ ×&Ñ&Ö(‰
ˆÜ�F C�L¨%Ö0ñ )à€Mr   N)Ú%torch.distributed.checkpoint.metadatar   Ú r   Ú	_traverser   r   r   r   r	   Údictr   r   r'   r.   © r   r   Ú<module>r4      s8   ðõ Bå ÷õ ðð �s˜H�}Õ%€õ ÷Fr   