΅i.%SSKrSSKJr SSKJr SSKJr SSKJr SSK J r SSK J r /r\\\S 'S \\4S jr\"S \S 9S\\S\S \4Sj5r\"S\S 9S\S\S\S\\\\\4S S4 Sj5rS\\S\ \S \ \4Sjr!S\S \4Sjr"S\S \4Sjr#S\S \\$\\\4\$\\4\ \\$\\\444Sjr%S\ \S \\$\\\4\$\\4\ \\$\\\444S jr&S;S\S!\S \\4S"jjr'S\S \\4S#jr(SN) defaultdict)Path)Any) deprecated)format_function_signature) FileManager__all__returnc/$Nr\/home/james-whalen/.local/lib/python3.13/site-packages/torch/utils/data/datapipes/gen_pyi.py__dir__rs Irzg`torch.utils.data.datapipes.gen_pyi.materialize_lines` is deprecated and will be removed in the future.)categorylines indentationc~SnSSU--n[U5H#upEUS:waX#- nX%RSU5- nM% U$)N  r) 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.dir template_name output_name replacementsc [RRX5n[RRX5n[USS9nUR 5nSSS5 UHFupn [USSS9nWR U[ X55nURU5 SSS5 MH g!,(df  N[=f!,(df  Mk=f)Nutf-8encodingw)ospathjoinopenreadrrwrite) r r!r"r# template_path output_pathfcontent placeholderrrs rgen_from_templater4.sGGLL4M'',,s0K mg .!&&( /+7' K +sW 5oo.uBG GGG  6 5,8 / .6 5s B/5-C/ B= C dir_pathsfiles_to_excludec `[5nUHn[R"U5nUVs1sHnSUSS:XdMUiM nnUVs1sH oUU;dM UiM nnURUVs1sH"n[RR X55iM$ sn5 M U$s snfs snfs snf)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)setr)listdirupdater*r+)r5r6pathsdir_path all_filesfname python_files filter_filess rfind_file_pathsrBEs eEJJx( +4L9%rs8K9 L+ +eT;Q!T)'++K8 A%  #S1F|!# JJsO3  JJsO3 #q(**S/Cds+I7H7S'48B59C6KY%)&Q-IC +N   M + *sCFBF F" file_pathsc0n0n[5n0nUHWn[U5unnnn URU5 URU5 URU5 URU 5 MY UUUU4$r )r9rhr;) rimethods_and_signaturesmethods_and_class_names!methods_with_special_output_typesmethods_and_doc_stringsr*r_r`$methods_needing_special_output_typesrbs rparse_datapipe_filesrps  (+%   %   0 %%&9:&&';<)001UV&&7 )  r delimitercSnSn/nUH?nUS:XaUS- nO,US:XaUS-nO XQ:XaUS:XaURU5 SnM;X5- nMA URU5 U$)zRGiven a line of text, split it on comma unless the comma is within a bracket '[]'.rr[rV])rZ)rrq bracket_count curr_tokenreschars rsplit_outside_bracketrysyMJ C 3; Q M S[ Q M  =A#5 JJz "J  JJz Jrcf[U5n[U5Hxup#URS5X'US:XaSX'M$US:aSXS- :XaXSS:waSX'MFSU;dMNURS 5SnURS5S -X'Mz UVs/sH oUS:wdM UPM nnU$s snf) z Clean up a given raw function signature. This includes removing the self-referential datapipe argument, default arguments of input functions, newlines, and spaces. rclsselfrrV*rz Callable ==z = ...)ryrr^ rpartition)rtokensrtokenheadts rr]r]s.d3Ff%KK$ E>FI U&Q-/filc6IFI U "##C(+D 3(2FI& +A7aF + M,s  B.$B.deprecated_filesdefault_output_typemethod_to_special_output_typerootcUS:Xa0[[[5RR 55n[ U[5(aU/OUnUVs/sH"n[ RRXV5PM$ nn[XRU5S9n[U5unn n n UHn X;dM U RU 5 M /n UR5HzupXnX;aXNnOUnSRX5nUS:XaSnOSU-n[XU5nU RSUS3UR!S5R#5-U-5 M| U R%SS 9 U $s snf) 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)r6z ...rz# Functional form of 'z' z...c*URS5S$)NrrV)split)ss r(get_method_definitions..saggdmA&r)key)strr__file__parentresolve isinstancer)r*r+rBunionrpr[itemsrrZ removesuffixrstripsort)rSr6rrrrr*rirkrlmethods_w_special_output_typesrnfn_namemethod_definitionsrd argumentsre 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)E+iteriterDP_file_pathz __init__.pyzutils.pyiterDP_files_to_excludeiterDP_deprecated_fileszlist[IterDataPipe])demuxfork$iterDP_method_to_special_output_typemapmapDP_file_pathmapDP_files_to_excludemapDP_deprecated_filesshuffle IterDataPipe#mapDP_method_to_special_output_typec^^[[[[S[5m[[ [ [S[5m[[5R5Rn[XSS9nURSSUU4Sj5 g) 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>TTS.$)N)IterDataPipeMethodsMapDataPipeMethodsr)iter_method_definitionsmap_method_definitionssrrmain..Hs#:"8 rN)rrrrrrrrrrrabsoluterrwrite_with_template)r*fmrrs @@rmainr,sz5, 4+  > " " $ + +D % HB r__main__),)r)r N)4r) collectionsrpathlibrtypingrtyping_extensionsrtorchgen.api.pythonrtorchgen.utilsr ImportErrorsysrrparents REPO_ROOTr*insertrrHr rX__annotations__r FutureWarningintrtupler4r9rBrNrRdictrhrpryr]rrrrrrrrrr__name__rrrrs #( =*cc m T#YSS  m  uS#s]+,    &tCyCHS" c c SS11 4T#Y c3hS4T#Y;O OP1hC 4T#Y c3hS4T#Y;O OP4d3i&CDI8 7T#Y7#h7#h7 7 $(S> 7  7 #Y7v#%2J$?S?$'ES) ! 8$d38n $1:#>C>#&5C(7@.6Q#T#s(^Q F zFG  X'')11!4IHHOOAs9~&=* 388}qSXXa[C N: HHQK s F--BII