+ j.6aR-t00t^RIt^RIHt^RIHt^RIHt^RIH t ^RI H t ^RI H t .t]^kRR lt] !R ]R 7R R l4t] !R]R 7RRl4tRRltRRltRRltRRlt RRlt!R.RRllt"RRlt#R/RR llt$R!t%]^kR"R#0t&]^k]'!4t(]^kR$R%R&R%/t)]^kR't*]^kR"R#0t+]^k]'!4t,]^kR(R)/t-]^kR*R+lt.]/R,8Xd ].!4R#R# ]d^RIt]!]4P#4P$^,t]P(P+^]!]44^RI H t ^RI H t ]!]P(4^8d,]P(^,]!]48Xd]P(^ELgi;i)0N) defaultdict)Path)Any) deprecated)format_function_signature) FileManagerc:V^8dQhR\\,/#)return)liststr)formats"c/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/utils/data/datapipes/gen_pyi.py __annotate__rscc.#Nrrr__dir__rs Irzg`torch.utils.data.datapipes.gen_pyi.materialize_lines` is deprecated and will be removed in the future.)categorycRV^8dQhR\\,R\R\/#)r lines indentationr )r r int)rs"rrr$s%T#YSSrcRpRRV,,p\V4F,wrEV^8wd W#, pW%PRV4, pK. V#)  ) enumeratereplace)rroutputnew_line_with_indentilines&& rmaterialize_linesr% sT F# "33U# 6  *F,,t%9::$ Mrzg`torch.utils.data.datapipes.gen_pyi.gen_from_template` is deprecated and will be removed in the future.c V^8dQhR\R\R\R\\\\\3,,RR/#)r dir template_name output_name replacementsr N)r r tuplerr)rs"rrr2sJ uS#s]+,   rc \PPW4p\PPW4p\VRR7;_uu_4pVP 4pRRR4VFQwrp \VRRR7;_uu_4pXP V\ W44pVPV4RRR4KS R# +'giLi;i +'giK{;i)utf-8encodingNw)ospathjoinopenreadr r%write) r'r(r)r* template_path output_pathfcontent placeholderrrs &&&& rgen_from_templater<.sGGLL4M'',,s0K mg . .!&&( /+7' K +sW 5 5oo.uBG GGG  6 5,8 / .6 5 5sC -C C  C* c~V^8dQhR\\,R\\,R\\,/#)r dir_pathsfiles_to_excluder )r r set)rs"rrrEs-tCyCHSrc ^\4pVFp\P!V4pVUu0uFpRVRR8XgKVkK ppVUu0uF qUV9gK VkK ppTPVUu0uF"p\PP W54kK$ up4K V#uupiuupiuupi)z When given a path to a directory, returns the paths to the relevant files within it. This function does NOT recursive traverse to subdirectories. z.pyN)r@r1listdirupdater2r3)r>r?pathsdir_path all_filesfname python_files filter_filess&& rfind_file_pathsrKEs eEJJx( +4L9%rs8K9 L+ +eT;Q!T)'++K8 A%  #S1F|!# JJsO3  JJsO3 #q(**S/Cds+I7H7S'48B59;RYKY%)&Q-IC +N   M + *s%;G .?G .G  AG 'BG  G c $V^8dQhR\\,R\\\\\,3,\\\3,\\,\\\\,3,3,/#)r file_pathsr )r@r r+r`r )rs"rrrsTC 4T#Y c3hS4T#Y;O OPrc/p/p\4p/pVFWp\V4wpppp VPV4VPV4VPV4VPV 4KY VVVV3#r)r@rrrD) rtmethods_and_signaturesmethods_and_class_names!methods_with_special_output_typesmethods_and_doc_stringsr2rirj$methods_needing_special_output_typesrls & rparse_datapipe_filesr{s  (+%   %   0 %%&9:&&';<)001UV&&7 )  rcRV^8dQhR\R\R\\,/#)r r$ delimiterr r r )rs"rrrs%d3irc^pRp.pVFOpVR8Xd V^, pM3VR8Xd V^,pM"WQ8XdV^8XdVPV4RpKGW5, pKQ VPV4V#)zRGiven a line of text, split it on comma unless the comma is within a bracket '[]'.r[])rd)r$r} bracket_count curr_tokenreschars&& rsplit_outside_bracketrsyMJ C 3; Q M S[ Q M  =A#5 JJz "J  JJz JrcFV^8dQhR\R\\,/#rMr~)rs"rrrsCDIrc\V4p\V4Fwr#VPR4W&VR8XdRW&K%V^8d/RW^, ,8XdW,^,R8wdRW&KZRV9gKcVPR4^,pVPR4R,W&K VUu.uF qUR8wgK VNK ppV#uupi) z Clean up a given raw function signature. This includes removing the self-referential datapipe argument, default arguments of input functions, newlines, and spaces. rclsself*rz Callable ==z = ...)rrrh rpartition)r$tokensr#tokenheadts& rrgrgs.d3Ff%KK$ E>FI U&Q-/filc6IFI U "##C(+D 3(2FI& +A7aaF + M,s 4C C cV^8dQhR\\\,,R\\,R\\,R\R\\\3,R\R\\,/#)r r_r?deprecated_filesdefault_output_typemethod_to_special_output_typerootr )r r r@r`)rs"rrrsj77T#Y7#h7#h7 7 $(S> 7  7 #Y7rc$VR8Xd1\\\4PP 44p\ V\4'dV.MTpVUu.uF"p\ PPWV4NK$ pp\WPV4R7p\V4wpp p p VFp W9gK V PV 4K .p VP4FwrW,pW9d WN,pMTpRPW,4pVR8XdRpM RV,p\WV4pV PRV R2VP!R4P#4,V,4K V P%RR 7V #uupi) z #.pyi generation for functional DataPipes Process. # 1. Find files that we want to process (exclude the ones who don't) # 2. Parse method name and signature # 3. Remove first argument after self (unless it is "*datapipes"), default args, and spaces r)r?z ...rz# Functional form of 'z' z...c2VPR4^,#)r)split)ss&r(get_method_definitions..saggdmA&r)key)r r__file__parentresolve isinstancer1r2r3rKunionr{reitemsrrd removesuffixrstripsort)r_r?rrrrr2rtrvrwmethods_w_special_output_typesryfn_namemethod_definitionsrn argumentsro output_type doc_string definitions&&&&&& rget_method_definitionsrs rz4>((0023))S99 yI6?@idd)iI@ $:$:;K$LJ Z( &1  8 * . .w 71"8">">"@ ,9  87DK-KWW4AB  J *J.{{S !!$ZL 4%%e,335 6  #A" & KAs(F iterz __init__.pyzutils.pydemuxzlist[IterDataPipe]forkmapshuffle IterDataPipecV^8dQhRR/#)r r Nr)rs"rrr,s  d rcaa\\\\R\4o\\ \ \R\4o\\4P4Pp\WRR7pVPRRVV3Rl4R#) a # Inject file into template datapipe.pyi.in. TODO: The current implementation of this script only generates interfaces for built-in methods. To generate interface for user-defined DataPipes, consider changing `IterDataPipe.register_datapipe_as_function`. r MapDataPipeF) install_dir template_dirdry_runz datapipe.pyizdatapipe.pyi.inc<RSRS/#)IterDataPipeMethodsMapDataPipeMethodsr)iter_method_definitionsmap_method_definitionssrrmain..Hs !#: "8 rN)riterDP_file_pathiterDP_files_to_excludeiterDP_deprecated_files$iterDP_method_to_special_output_typemapDP_file_pathmapDP_files_to_excludemapDP_deprecated_files#mapDP_method_to_special_output_typerrabsoluterrwrite_with_template)r2fmrrs @@rmainr,sz5, 4+  > " " $ + +D % HB r__main__c2V^8dQh/^\9d\\,;R&^\9d \;R&^\9d\\,;R&^\9d\\,;R&^\9d\\\3,;R&^\9d \;R&^\9d\\,;R&^\9d\\,;R&^\9d\\\3,;R &#) r __all__rrrrrrrr)__conditional_annotations__r r r@r`)rs"rrrs .c/ x#y z@?S?{ |*)S)} ~d38n H I  J ?>C>K  L )(C(M  N RQT#s(^QO  r),)r)1rr1 collectionsrpathlibrtypingrtyping_extensionsrtorchgen.api.pythonrtorchgen.utilsr ImportErrorsysrrparents REPO_ROOTr2insertr rRrr FutureWarningr%r<rKrXr]rrr{rrgrrrr@rrrrrrr__name__r)rs@rrs #( =* m   m  &" 1h4&,7v%2J$??$'E) ! 8$ $1:#>>#&5(7@.6Q#Q F zFG  X'')11!4IHHOOAs9~&=* 388}qSXXa[C N: HHQK s C22B"FF