ó
    Eñi  ã            	       óê  • % S SK Jr  S SKJr  S SKJrJrJr  S SKJ	r	  S SK
rS SKJrJrJr  \\\/\4   r\\\/\4   r0 r\\\   \4   \S'   0 r\\\   \\   4   \S'   \" S	5      r\" S
5      r\" S5      r SS\\   S\S\\   SS4S jjrS\\   SS4S jrS\S\S\\   4S jrS\\\4   S\S\\   4S jrS\\   S\S\\   4S jr S\!\S4   S\S\\   4S jr"S\	S\S\\   4S jr#S\\\4   S\S\4S jr$S\\   S\S\4S jr%S\!\S4   S\S\4S jr&S\	S\S\4S jr'\" \\\$5        \" \\ \%5        \" \!\"\&5        \RP                  RR                   H  r*\" \*\"\&5        M     \" \\#\'5        g) é    )Ú
namedtuple)ÚCallable)ÚAnyÚOptionalÚTypeVar)Ú
NamedTupleN)ÚPyTreeÚtree_flattenÚTreeSpecÚSUPPORTED_NODESÚSUPPORTED_NODES_EXACT_MATCHÚ_TÚ_KÚ_VÚclsÚflatten_fn_specÚflatten_fn_exact_match_specÚreturnc                 ó(   • U[         U '   U[        U '   g ©N©r   r   )r   r   r   s      ÚM/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/fx/_pytree.pyÚregister_pytree_flatten_specr      s   € ð
 +„O�CÑØ'BÔ Ò$ó    c                 ó    • [         U 	 [        U 	 g r   r   )r   s    r   Ú_deregister_pytree_flatten_specr      s   € ô 	˜ÐÜ# CÑ(r   ÚpytreeÚspecc           	      óL  • UR                  5       (       a  U /$ UR                  [        ;   aP  [        UR                     nU" X5      n/ n[        X1R	                  5       5       H  u  pV[        XV5      nXG-  nM     U$ [        U 5      u  p‰X:w  a  [        SU	 SU  SU S35      eU$ )Nz
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_leafÚtyper   Ú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   Údc                 óJ   • UR                    Vs/ s H  o U   PM	     sn$ s  snf r   )Úcontext)r-   r   Úks      r   Ú_dict_flatten_specr1   A   s    € ØŸ,š,Ó'š,�QˆaŒD™,Ñ'Ð'ùÒ's   � c                 ó\   • [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   ©ÚrangeÚnum_children©r-   r   Úis      r   Ú_list_flatten_specr8   E   ó)   € Ü × 1Ñ 1Ô2Ó3Ò2�QˆaŒDÑ2Ñ3Ð3ùÒ3ó   ˜).c                 ó\   • [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   r3   r6   s      r   Ú_tuple_flatten_specr<   I   r9   r:   c                 ó\   • [        UR                  5       Vs/ s H  o U   PM	     sn$ s  snf r   r3   r6   s      r   Ú_namedtuple_flatten_specr>   M   r9   r:   c                 ó2   • [        U 5      UR                  :H  $ r   ©Úlenr5   ©r-   r   s     r   Ú_dict_flatten_spec_exact_matchrC   Q   ó   € Üˆq‹6�T×&Ñ&Ñ&Ð&r   c                 ó2   • [        U 5      UR                  :H  $ r   r@   rB   s     r   Ú_list_flatten_spec_exact_matchrF   U   rD   r   c                 ó2   • [        U 5      UR                  :H  $ r   r@   rB   s     r   Ú_tuple_flatten_spec_exact_matchrH   Y   rD   r   c                 ó2   • [        U 5      UR                  :H  $ r   r@   rB   s     r   Ú$_namedtuple_flatten_spec_exact_matchrJ   ]   rD   r   r   )+Úcollectionsr   Úcollections.abcr   Útypingr   r   r   Útyping_extensionsr   Útorch.return_typesÚtorchÚtorch.utils._pytreer	   r
   r   ÚlistÚFlattenFuncSpecÚboolÚFlattenFuncExactMatchSpecr   Údictr!   Ú__annotations__r   r   r   r   r   r   r$   r1   r8   Útupler<   r>   rC   rF   rH   rJ   Úreturn_typesÚall_return_typesÚreturn_type© r   r   Ú<module>r]      sˆ  ðÞ "Ý $ß )Ñ )Ý (ã ß >Ñ >ð ˜F HÐ-¨tÐ3Ñ4€Ø$ f¨hÐ%7¸Ð%=Ñ>Ð à46€��d˜3‘i Ð0Ñ1Ó 6ØTVÐ ˜T $ s¡)¨XÐ6OÑ-PÐ"PÑQÓ VáˆTƒ]€ÙˆTƒ]€ÙˆTƒ]€ð HLñCØ	ˆc‰ðCà$ðCð "*Ð*CÑ!DðCð 
õ	Cð)Ø	ˆc‰ð)à	ô)ðØðà
ðð 
ˆ#�Yôð8(˜$˜r 2˜v™,ð (¨hð (¸4À¹8ô (ð4˜$˜r™(ð 4¨(ð 4°t¸B±xô 4ð4˜5  S ™>ð 4°ð 4¸dÀ2¹hô 4ð4 
ð 4°(ð 4¸tÀC¹yô 4ð' d¨2¨r¨6¡lð '¸(ð 'Àtô 'ð' d¨2¡hð '°hð 'À4ô 'ð' u¨R°¨W¡~ð '¸Xð 'È$ô 'ð'¨Jð '¸hð 'È4ô 'ñ ˜TÐ#5Ð7UÔ VÙ ˜TÐ#5Ð7UÔ VÙ Ø	ØØ#ôð
 ×%Ñ%×6Ô6€KÙ ØØØ'öñ 7ñ ØØØ(õr   