+
    &j                         ^ RI t ^ RIt^ RIt^ RIHt ^ RIHt ^ RIHtH	t	 RR.t
]]	,          t]]]]R3,          3,          tR R	 ltR
 R ltR R ltRR R lltR R ltR# )    N)
Collection)dill_available)IterDataPipeMapDataPipetraversetraverse_dpsDataPipeGraphc                $    V ^8  d   QhR\         /# )   return)str)formats   "n/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/utils/data/graph.py__annotate__r      s           c                      R # )STUB r   r   r   _stub_unpicklerr      s    r   c          	      t    V ^8  d   QhR\         R\        R\        \        ,          R\        \         ,          /# )r   scan_objonly_datapipecacher   )DataPipeboolsetintlist)r   s   "r   r   r      s:     A  A A '+A 47HA 	(^A r   c                 N  a aa \         P                  ! 4       p\        P                  ! V4      p\	        4       '       d   ^ RIHp V! V4      pMRp. oR pVVV 3R lp\        \        3p	 V	 F/  p
V
P                  V4       V'       g   K  V
P                  V4       K1  	   VP                  S 4       T	 F/  p
T
P                  R4       T'       g   K  T
P                  R4       K1  	  \	        4       '       d   ^ RIHp T! R4       S#   \        P                  \        \        3 d&    \	        4       '       d   TP                  S 4        Lh i ; i  T	 F/  p
T
P                  R4       T'       g   K  T
P                  R4       K1  	  \	        4       '       d   ^ RIHp T! R4       i i ; i)r   )PicklerNc                    R p\        V \        4      '       dE   / pV P                  4        F,  w  r#\        V\        \        \
        34      '       g   K(  W1V&   K.  	  V# \        V \        \        34      '       dB   . pV  F7  p\        V\        \        \
        34      '       g   K&  VP                  V4       K9  	  V# \        V \        \        \
        34      '       d   T pV# N)	
isinstancedictitemsr   r   r   tupler   append)	ori_statestatekvs   &   r   getstate_hook0_list_connected_datapipes.<locals>.getstate_hook'   s    i&&E!)a,Z!HII !H *  	E4=11Ea,Z!HIILLO 
  	L+z#JKKEr   c                    < V S8X  g   \        V 4      S9   d   \        hSP                  V 4       SP                  \        V 4      4       \        R3# )Nr   )idNotImplementedErrorr'   addr   )objr   captured_connectionsr   s   &r   reduce_hook._list_connected_datapipes.<locals>.reduce_hook7   sD    (?bg.%% '',IIbg"B&&r   )extendF)ioBytesIOpickler    r   dillr   r   set_reduce_ex_hookset_getstate_hookdumpPickleErrorAttributeError	TypeErrorr6   )r   r   r   fpdill_Picklerdr,   r4   datapipe_classesclsdill_extendr3   s   f&f         @r   _list_connected_datapipesrH      sc    	

A		A 0O ' 0<[.I#C"";/}%%m4 $	FF8 $C""4(}%%d+ $ 2 ""NI> 	x 	 $C""4(}%%d+ $ 2 s7   *E 
E  D AEE 	EE F$/5F$c                0    V ^8  d   QhR\         R\        /# )r   datapiper   )r   r	   )r   s   "r   r   r   [   s     G G8 G Gr   c                2    \        4       p\        V RVR7      # )a  
Traverse the DataPipes and their attributes to extract the DataPipe graph.

This only looks into the attribute from each DataPipe that is either a
DataPipe and a Python collection object such as ``list``, ``tuple``,
``set`` and ``dict``.

Args:
    datapipe: the end DataPipe of the graph
Returns:
    A graph represented as a nested dictionary, where keys are ids of DataPipe instances
    and values are tuples of DataPipe instance and the sub-graph
T)r   r   )r   _traverse_helper)rJ   r   s   & r   r   r   [   s     eEHDFFr   c                J    V ^8  d   QhR\         R\        R,          R\        /# )r   rJ   r   Nr   )r   r   r	   )r   s   "r   r   r   m   s%     < <x <t <} <r   c                    RpV'       g
   VR,          p\         P                  ! V\        ^R7       Vf   Rp\        4       p\	        WV4      # )a  
Traverse the DataPipes and their attributes to extract the DataPipe graph.

[Deprecated]
When ``only_dataPipe`` is specified as ``True``, it would only look into the
attribute from each DataPipe that is either a DataPipe and a Python collection object
such as ``list``, ``tuple``, ``set`` and ``dict``.

Note:
    This function is deprecated. Please use `traverse_dps` instead.

Args:
    datapipe: the end DataPipe of the graph
    only_datapipe: If ``False`` (default), all attributes of each DataPipe are traversed.
      This argument is deprecating and will be removed after the next release.
Returns:
    A graph represented as a nested dictionary, where keys are ids of DataPipe instances
    and values are tuples of DataPipe instance and the sub-graph
zV`traverse` function and will be removed after 1.13. Please use `traverse_dps` instead.zM And, the behavior will be changed to the equivalent of `only_datapipe=True`.)
stacklevelF)warningswarnFutureWarningr   rL   )rJ   r   msgr   s   &&  r   r   r   m   sK    *	-  ^^MM#}3eEHU;;r   c                ^    V ^8  d   QhR\         R\        R\        \        ,          R\        /# )r   rJ   r   r   r   )r   r   r   r   r	   )r   s   "r   r   r      s0      '+47Hr   c           	      r   \        V \        \        34      '       g   \        R \	        V 4       R24      h\        V 4      pW29   d   / # VP                  V4       \        WVP                  4       4      pW0/ 3/pV F9  pWS,          ^,          P                  \        WaVP                  4       4      4       K;  	  V# )z.Expected `IterDataPipe` or `MapDataPipe`, but z	 is found)r#   r   r   RuntimeErrortyper/   r1   rH   copyupdaterL   )rJ   r   r   dp_idr%   rD   items   &&&    r   rL   rL      s     h{ ;<<<T(^<LIV
 	
 xLE~		IIe%huzz|LE"~.A 	
+DNO	 
 Hr   r"   )r7   r9   rP   collections.abcr   torch.utils._import_utilsr   #torch.utils.data.datapipes.datapiper   r   __all__r   r$   r   r&   r	   r   rH   r   r   rL   r   r   r   <module>r`      sc    	   & 4 I ~
&+%S%/ 9::;
A HG$<Dr   