+ j^RIt^RIt^RIHt^RIHt^RIt^RIHtH t ^RI H t H t H t .ROtRRltRR ltR R lt] P$3R R lltRRltRRRllt]!R]R7RRl4tRRltRRltR#)N)Any) deprecated)_ShardingIterDataPipeSHARDING_PRIORITIES)DataPipe DataPipeGraph traverse_dpscFV^8dQhR\R\\,/#)graphreturn)rlistr)formats"`/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/utils/data/graph_settings.py __annotate__rs55}5h5c*\V\44#N)_get_all_graph_pipes_helperset)r s&rget_all_graph_pipesrs &uce 44rchV^8dQhR\R\\,R\\,/#)r r id_cacher )rrintrr)rs"rrrs,    $'H  (^ rc.pVP4FLwpwrEW19dKVPV4VPV4VP\ WQ44KN V#r)itemsaddappendextendr)r rresultsdp_iddatapipe sub_graphs&& rrrs[!G(- $$    Ux 29GH )6 Nrc0V^8dQhR\R\/#r r"r rbool)rs"rrr)sHrc\V\4;'g5\VR4;'d!\P!VP 4#)apply_sharding) isinstancerhasattrinspectismethodr)r"s&r_is_sharding_datapiper/)sE h 5 6  *+ 6 6   X44 5rcHV^8dQhR\R\R\R\/#)r r"num_of_instances instance_idr )rr)rs"rrr0s0%%%%%  %rcPaaaa\V4pRRVVVV3RllloS!V4V#)z Apply dynamic sharding over the ``sharding_filter`` DataPipe that has a method ``apply_sharding``. RuntimeError will be raised when multiple ``sharding_filter`` are presented in the same branch. cV^8dQhRR/#)r r N)rs"rr$apply_sharding..__annotate__=s((T(rcl<VP4Fwr#Rp\V4'dyVe\RV RV 24h\P!VP 4p\ VP4^8dVP SS4MVP SSS R7TpVfTpS!W44K R#)NzoSharding twice on a single pipeline is likely unintended and will cause data loss. Sharding already applied to z while trying to apply to )sharding_group)valuesr/ RuntimeErrorr, signaturer)len parameters) r prev_applieddpr#appliedsig_helperr2r1r8s && rrBapply_sharding.._helper=s"\\^MBG$R((+&77CnD^_a^bd ''(9(9:s~~&*%%&6 D%%(+n&& I '',rr)r )r"r1r2r8r rBs&fff @rr)r)0s'  "E((, EN Orc0V^8dQhR\R\/#r%r&)rs"rrrXs8rc\VR4;'d]\VR4;'dI\P!VP4;'d!\P!VP4#) set_shuffleset_seed)r+r,r-rFrGr.s&r_is_shuffle_datapiperHXsc-( 0 0 Hj ) 0 0   X11 2 0 0   X.. / rcJV^8dQhR\R\R,R\/#)r r"shuffleNr r&)rs"rrras%Xt xrc:VfV#\V4p\V4pVUu.uFp\V4'gKVNK ppV'g4V'd,\P!R^R7VP 4pV.pVFpVP V4K V#uupi)a) Traverse the graph of ``DataPipes`` to find and set shuffle attribute. Apply the method to each `DataPipe` that has APIs of ``set_shuffle`` and ``set_seed``. Args: datapipe: DataPipe that needs to set shuffle attribute shuffle: Shuffle option (default: ``None`` and no-op to the graph) z`shuffle=True` was set, but the datapipe does not contain a `Shuffler`. Adding one at the end. Be aware that the default buffer size might not be sufficient for your task.) stacklevel)r rrHwarningswarnrJrF)r"rJr all_pipespipe shufflersshufflers&& rapply_shuffle_settingsrSas  "E#E*I"+J)$/CD/I)IJ   [ ##%  W% OKs BBz`apply_shuffle_seed` is deprecated since 1.12 and will be removed in the future releases. Please use `apply_random_seed` instead.)categoryc<V^8dQhR\R\R\/#r r"rngr )rr)rs"rrrs!,,,,,rc\W4#r)apply_random_seed)r"rWs&&rapply_shuffle_seedrZs X ++rc0V^8dQhR\R\/#r%r&)rs"rrrsQQ(QtQrcj\VR4;'d!\P!VP4#)rG)r+r,r-rGr.s&r_is_random_datapiper]s( 8Z ( P PW-=-=h>O>O-PPrcPV^8dQhR\R\PR\/#rV)rtorch Generator)rs"rrrs%u8rc\V4p\V4p\4p.pVFSp\V4V9dK\ V4'gK(VP V4VP \V44KU VFap\\P!R\PR7PVR7P44pVPV4Kc V#)a Traverse the graph of ``DataPipes`` to find random ``DataPipe`` with an API of ``set_seed``. Then set the random seed based on the provided RNG to those ``DataPipe``. Args: datapipe: DataPipe that needs to set randomness rng: Random number generator to generate random seeds )dtype) generatorr5)r rridr]rrrr_emptyint64random_itemrG)r"rWr rOcacherandom_datapipesrP random_seeds&& rrYrYs  "E#E*I EE d8u   t $ $  # #D ) IIbh  ! KK%++ . 6 6 6 E J J L  k" ! Or)rYr)rZrSrr)r,rMtypingrtyping_extensionsrr_(torch.utils.data.datapipes.iter.shardingrrtorch.utils.data.graphrrr __all__rrr/DEFAULTr)rHrS FutureWarningrZr]rYr5rrrss( IH 5 '.. %PD .  ,  ,Qr