+
    –°³j²  ã                   ó  a € R  t+0 t ^ RIHt ^ RIHt ^ RIHtHt ^ RIH	t	 ^ RI
t^ RIHtHtHt ]]].]],          3,          t]]].]3,          t]t]t/ t] ^ k / t] ^k ]! R4      t]! R4      t]! R	4      tR!R
 R lltR R ltR R ltR R ltR R ltR R lt R R lt!R R lt"R R lt#R R lt$R R lt%]! ]&]]"4       ]! ]]]#4       ]! ]'] ]$4       ]PP                  PR                   F  t*]! ]*] ]$4       K  	  ]! ]]!]%4       R# )"é    )Ú
namedtuple)ÚCallable)ÚAnyÚTypeVar)Ú
NamedTupleN)ÚPyTreeÚtree_flattenÚTreeSpecÚ_TÚ_KÚ_Vc                ód   € V ^8„  d   QhR\         \        ,          R\        R\        R,          RR/# )é   ÚclsÚflatten_fn_specÚflatten_fn_exact_match_specNÚreturn)Útyper   ÚFlattenFnSpecÚFlattenFnExactMatchSpec)Úformats   "ÚQ/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/fx/_pytree.pyÚ__annotate__r      s?   € ÷ Cñ CÜ	Œc�ðCä"ðCô "9¸4Õ!?ðCð 
ñ	Có    c                 ó*   € V\         V &   V\        V &   R # ©N©ÚSUPPORTED_NODESÚSUPPORTED_NODES_EXACT_MATCH)r   r   r   s   &&&r   Úregister_pytree_flatten_specr       s   € ð
 +„O�CÑØ'BÔ Ó$r   c                ó>   € V ^8„  d   QhR\         \        ,          RR/# )r   r   r   N)r   r   )r   s   "r   r   r   "   s    € ÷ )ñ )Ü	Œc�ð)à	ñ)r   c                 ó"   € \         V  \        V  R # r   r   )r   s   &r   Ú_deregister_pytree_flatten_specr#   "   s   € ô 	˜ÐÜ# CÒ(r   c                óR   € V ^8„  d   QhR\         R\        R\        \        ,          /# )r   ÚpytreeÚspecr   )r   r
   Úlistr   )r   s   "r   r   r   )   s*   € ÷ ñ Üðä
ðô 
Œ#…Yñr   c           	      ób  € VP                  4       '       d   V .# VP                  \        9   dY   \        VP                  ,          pV! W4      p. p\        W1P	                  4       4       F  w  rV\        WV4      pWG,          pK  	  V# \        V 4      w  r‰W8w  d   \        R V	 RV  RV R24      hV# )z
Real spec z of object z! is different from expected spec z_. Please file an issue at https://github.com/pytorch/pytorch/issues/new?template=bug-report.yml)Úis_leafr   r   ÚzipÚchildrenÚtree_flatten_specr	   ÚRuntimeError)
r%   r&   r   Úchild_pytreesÚresultÚchildÚ
child_specÚflatÚflat_resultÚ	real_specs
   &&        r   r,   r,   )   s½   € ð ‡|�|‡~‚~Øˆxˆð ‡y�y”OÔ#Ü)¨$¯)©)Õ4ˆÙ'¨Ó5ˆØˆÜ!$ ]·M±M³OÖ!DÑˆEÜ$ UÓ7ˆDØ�NŠFñ "Eð ˆÜ)¨&Ó1Ñ€KØÔÜØ˜˜ ;¨v¨hÐ6WÐX\ÐW]ð ^lð mó
ð 	
ð Ðr   c                ót   € V ^8„  d   QhR\         \        \        3,          R\        R\        \        ,          /# ©r   Údr&   r   )Údictr   r   r
   r'   )r   s   "r   r   r   E   s-   € ÷ (ñ (œ$œr¤2˜v�,ð (¬hð (¼4Ä½8ñ (r   c                 óP   € VP                    Uu. uF  q V,          NK  	  up# u upi r   )Úcontext)r7   r&   Úks   && r   Ú_dict_flatten_specr<   E   s"   € ØŸ,š,Ó'™,�Qˆa�DˆD™,Ñ'Ð'ùÒ's   �#c                óh   € V ^8„  d   QhR\         \        ,          R\        R\         \        ,          /# r6   )r'   r   r
   )r   s   "r   r   r   I   s)   € ÷ 4ñ 4œ$œr�(ð 4¬(ð 4´t¼Bµxñ 4r   c                 ób   € \        VP                  4       Uu. uF  q V,          NK  	  up# u upi r   ©ÚrangeÚnum_children©r7   r&   Úis   && r   Ú_list_flatten_specrD   I   ó+   € Ü × 1Ñ 1Ô2Ó3Ñ2�Qˆa�DˆDÑ2Ñ3Ð3ùÒ3ó   ˜,c                ól   € V ^8„  d   QhR\         \        R3,          R\        R\        \        ,          /# ©r   r7   .r&   r   )Útupler   r
   r'   )r   s   "r   r   r   M   s-   € ÷ 4ñ 4œ5¤ S �>ð 4´ð 4¼dÄ2½hñ 4r   c                 ób   € \        VP                  4       Uu. uF  q V,          NK  	  up# u upi r   r?   rB   s   && r   Ú_tuple_flatten_specrK   M   rE   rF   c                óR   € V ^8„  d   QhR\         R\        R\        \        ,          /# r6   )r   r
   r'   r   )r   s   "r   r   r   Q   s%   € ÷ 4ñ 4¤
ð 4´(ð 4¼tÄC½yñ 4r   c                 ób   € \        VP                  4       Uu. uF  q V,          NK  	  up# u upi r   r?   rB   s   && r   Ú_namedtuple_flatten_specrN   Q   rE   rF   c                ó^   € V ^8„  d   QhR\         \        \        3,          R\        R\        /# r6   )r8   r   r   r
   Úbool)r   s   "r   r   r   U   s)   € ÷ 'ñ '¤d¬2¬r¨6¥lð '¼(ð 'Ätñ 'r   c                 ó2   € \        V 4      VP                  8H  # r   ©ÚlenrA   ©r7   r&   s   &&r   Ú_dict_flatten_spec_exact_matchrU   U   ó   € Üˆq‹6�T×&Ñ&Ñ&Ð&r   c                óR   € V ^8„  d   QhR\         \        ,          R\        R\        /# r6   )r'   r   r
   rP   )r   s   "r   r   r   Y   s%   € ÷ 'ñ '¤d¬2¥hð '´hð 'Ä4ñ 'r   c                 ó2   € \        V 4      VP                  8H  # r   rR   rT   s   &&r   Ú_list_flatten_spec_exact_matchrY   Y   rV   r   c                óV   € V ^8„  d   QhR\         \        R3,          R\        R\        /# rH   )rI   r   r
   rP   )r   s   "r   r   r   ]   s)   € ÷ 'ñ '¤u¬R°¨W¥~ð '¼Xð 'Ì$ñ 'r   c                 ó2   € \        V 4      VP                  8H  # r   rR   rT   s   &&r   Ú_tuple_flatten_spec_exact_matchr\   ]   rV   r   c                ó<   € V ^8„  d   QhR\         R\        R\        /# r6   )r   r
   rP   )r   s   "r   r   r   a   s!   € ÷ 'ñ '¬Jð '¼hð 'Ì4ñ 'r   c                 ó2   € \        V 4      VP                  8H  # r   rR   rT   s   &&r   Ú$_namedtuple_flatten_spec_exact_matchr_   a   rV   r   c                óæ   € V ^8„  d   Qh/ ^ \         9   d&   \        \        \        ,          \        3,          ;R&   ^\         9   d-   \        \        \        ,          \
        R,          3,          ;R&   # )r   r   Nr   )Ú__conditional_annotations__r8   r   r   r   r   )r   s   "r   r   r      sO   € × "Ñ "÷  5Ò 4””dœ3•i¤Ð.Õ/Ñ 4ñ! #÷" RÒ QœT¤$¤s¥)Ô-DÀtÕ-KÐ"KÕLÑ Qò# #r   r   ),ra   Úcollectionsr   Úcollections.abcr   Útypingr   r   Útyping_extensionsr   Útorch.return_typesÚtorchÚtorch.utils._pytreer   r	   r
   r'   r   rP   r   ÚFlattenFuncSpecÚFlattenFuncExactMatchSpecr   r   r   r   r   r    r#   r,   r<   rD   rK   rN   rU   rY   r\   r_   r8   rI   Úreturn_typesÚall_return_typesÚreturn_typer   )ra   s   @r   Ú<module>rn      s)  øðß "Ò "Ý $ß Ý (ã ß >Ñ >ð ˜& (Ð+¨T°#­YÐ6Õ7€Ø" F¨HÐ#5°tÐ#;Õ<Ð ð  €Ø3Ð à24€Ó 4ØOQÐ Ó QáˆTƒ]€ÙˆTƒ]€ÙˆTƒ]€÷Cõ)õõ8(õ4õ4õ4õ'õ'õ'õ'ñ ˜TÐ#5Ð7UÔ VÙ ˜TÐ#5Ð7UÔ VÙ Ø	ØØ#ôð
 ×%Ñ%×6Ô6€KÙ ØØØ'öñ 7ñ ØØØ(ör   