§ ž²ÅjGãó—UdZddlmZddlmZmZmZmZddlm cm Zddlm Z erddlmZgZded <ejrÊddlZddlZddlm cmZe ejjd ¬¦«d d„¦«ZdZdD]*Zeee¦«Ze ed ¬¦«ej¦«[Œ+[e ejd ¬¦« d!d"d„¦«Ze ejd¬¦« d!d#d„¦«Zedgz Ze ej d ¬¦« d!d$d„¦«Z edgz ZdSdS)%z) Python polyfills for torch.utils.pytree é)Úannotations)ÚAnyÚCallableÚIterableÚ TYPE_CHECKINGNé)Úsubstitute_in_graph)ÚPyTreez list[str]Ú__all__T)Úcan_constant_fold_throughÚargsrÚkwargsÚreturnÚboolcó —td¦«‚)NzeShould not be called directly because the original function will be called in the constant fold path.)Ú ValueError)r rs úOC:\Projects\Historical\venv\Lib\site-packages\torch/_dynamo/polyfills/pytree.pyÚ_rs€õð Vñ ô ð óÚ)Ú is_namedtupleÚis_namedtuple_classÚis_namedtuple_instanceÚis_structseqÚis_structseq_classÚis_structseq_instanceÚnamedtuple_fieldsÚstructseq_fieldsÚtreer Úis_leafúCallable[[PyTree], bool] | Nonecóˆ—| | ||¦«rdStj t|¦«d¬¦«€dSdS)NTÚtorch)Ú namespaceF)ÚoptreeÚregister_pytree_nodeÚgetÚtype©rr s rÚtree_is_leafr*9sL€ð ˆ<˜GÐ/°G°G¸D±M´MÐ/Ø4ÝÔ&×*Ò*4°©:¬:ÀÐ*ÑIÔIÐQØ4ØˆurFú Iterable[Any]c#óèK—|g}|rj| ¦«}t||¬¦«r|V—Œ,tj||dd¬¦«^}}| t|¦«¦«|°hdSdS)N©r Tr#)r Únone_is_leafr$)Úpopr*r%Útree_flatten_one_levelÚextendÚreversed)rr ÚstackÚnodeÚchildrenrs rÚ tree_iterr6Ds¦èè€ð ˆØð -Ø—9’9‘;”;ˆDÝ˜D¨'Ð2Ñ2Ô2ð Ø Øå!Ô8ØØØ!Ø!ð ñôˆLˆHqð LŠL (Ñ+Ô+Ñ,Ô,Ð,ðð -ð -ð -ð -ð -rr6ú list[Any]có>—tt||¬¦«¦«S)Nr-)Úlistr6r)s rÚtree_leavesr:Zs€õ •I˜d¨GÐ4Ñ4Ô4Ñ5Ô5Ð5rr:)r rrrrr)N)rr r r!rr)rr r r!rr+)rr r r!rr7)!Ú__doc__Ú __future__rÚtypingrrrrÚtorch.utils._pytreeÚutilsÚ_pytreeÚ python_pytreeÚ decoratorsr Útorch.utils._cxx_pytreer rÚ__annotations__Ú_cxx_pytree_dynamo_traceabler%Ú optree._CÚ_cxx_pytreeÚ cxx_pytreeÚ_CÚis_dict_insertion_orderedrÚ__nameÚgetattrÚ__funcÚ__python_implementation__r*r6r:©rrúrPsjððððð#Ð"Ð"Ð"Ð"Ð"à9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9à+Ð+Ð+Ð+Ð+Ð+Ð+Ð+Ð+à,Ð,Ð,Ð,Ð,Ð,ðð/Ø.Ð.Ð.Ð.Ð.Ð.ð€ÐÐÐÑðÔ-ðLØ€M€M€MØÐÐÐà0Ð0Ð0Ð0Ð0Ð0Ð0Ð0Ð0àÐØŒ Ô+Ø"&ðñôð ð ð ñ ôð ð€Fð ððˆð˜ Ñ(Ô(ˆØCÐÐ˜F¸dÐCÑCÔCØÔ,ñ ô ð ð ˆFØàÐ˜Ô0ÈDÐQÑQÔQð48ððððñRÔQððÐ˜Ô-ÈÐOÑOÔOð48ð-ð-ð-ð-ñPÔOð-ð&ˆ}Ñ€GàÐ˜Ô/È4ÐPÑPÔPð48ð6ð6ð6ð6ñQÔPð6ð ˆÑ€G€G€GðYLðLr