+
    &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   "h/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/fx/_pytree.py__annotate__r      s?     C C	cC"C "94!?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    
 +OC'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  rW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 yyO#)$))4'5!$]MMO!DE$U7DNF "E )&1K;vh6WX\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-     ( ($r2v, (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"    ,,',QaDD,'''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( 4tBx 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2QaDD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 45S> 4 4d2h 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( 4tCy 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)     ' 'd2r6l '( 't 'r   c                 2    \        V 4      VP                  8H  # r   lenrA   r7   r&   s   &&r   _dict_flatten_spec_exact_matchrU   U       q6T&&&&r   c                R    V ^8  d   QhR\         \        ,          R\        R\        /# r6   )r'   r   r
   rP   )r   s   "r   r   r   Y   s%     ' 'd2h '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)     ' 'uR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d3i./ 4! #" R QT$s)-Dt-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"FH#5t#;<   3 24 4OQ  QT]T]T]C)8(444'''' T#57U V T#57U V 	#
 %%66K ' 7 (r   