tj jUddlmZddlmZddlmZmZddlmZddl Z ddl m Z m Z mZee egeefZee egefZeZeZiZeeeefed<iZeeeedzfed<ed Zed Zed Z dd eed ededzddfdZd eeddfdZde dedeefdZdeeefdedeefdZ deededeefdZ!de"edfdedeefdZ#dededeefdZ$deeefdedefdZ%deededefdZ&de"edfdedefdZ'dededefdZ(eee e%eee!e&ee"e#e'e j)j*D]Z+ee+e#e'eee$e(dS) ) namedtuple)Callable)AnyTypeVar) NamedTupleN)PyTree tree_flattenTreeSpecSUPPORTED_NODESSUPPORTED_NODES_EXACT_MATCH_T_K_Vclsflatten_fn_specflatten_fn_exact_match_specreturnc.|t|<|t|<dSNr r )rrrs N/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/fx/_pytree.pyregister_pytree_flatten_specrs +OC'B$$$c&t|=t|=dSrr)rs r_deregister_pytree_flatten_specr"s #C(((rpytreespecc p|r|gS|jtvr_t|j}|||}g}t||D]\}}t ||}||z }|St |\}} || krtd| d|d|d|S)Nz Real spec z of object z! is different from expected spec z_. Please file an issue at https://github.com/pytorch/pytorch/issues/new?template=bug-report.yml)is_leaftyper zipchildrentree_flatten_specr RuntimeError) rrr child_pytreesresultchild child_specflat flat_result real_specs rr#r#)s ||~~x  yO##)$)4'55 !$]DMMOO!D!D   E:$UJ77D dNFF )&11K y m m mv m mX\ m m m    rdc*fd|jDS)Nc g|] }| Sr/).0kr,s r z&_dict_flatten_spec..Fs ' ' 'QAaD ' ' 'r)contextr,rs` r_dict_flatten_specr5Es ' ' ' '$, ' ' ''rcDfdt|jDS)Nc g|] }| Sr/r/r0ir,s rr2z&_list_flatten_spec..J 3 3 3QAaD 3 3 3rrange num_childrenr4s` r_list_flatten_specr>I( 3 3 3 3% 122 3 3 33r.cDfdt|jDS)Nc g|] }| Sr/r/r8s rr2z'_tuple_flatten_spec..Nr:rr;r4s` r_tuple_flatten_specrBMr?rcDfdt|jDS)Nc g|] }| Sr/r/r8s rr2z,_namedtuple_flatten_spec..Rr:rr;r4s` r_namedtuple_flatten_specrEQr?rc2t||jkSrlenr=r4s r_dict_flatten_spec_exact_matchrIU q66T& &&rc2t||jkSrrGr4s r_list_flatten_spec_exact_matchrLYrJrc2t||jkSrrGr4s r_tuple_flatten_spec_exact_matchrN]rJrc2t||jkSrrGr4s r$_namedtuple_flatten_spec_exact_matchrParJrr), collectionsrcollections.abcrtypingrrtyping_extensionsrtorch.return_typestorchtorch.utils._pytreerr r list FlattenFnSpecboolFlattenFnExactMatchSpecFlattenFuncSpecFlattenFuncExactMatchSpecr dictr __annotations__r r rrrrr#r5r>tuplerBrErIrLrNrP return_typesall_return_types return_typer/rrrds"""""""$$$$$$((((((>>>>>>>>>>&(+T#Y67 "FH#5t#;< 324d3i./444OQT$s)-Dt-K"KLQQQ WT]] WT]] WT]] CGCC cC"C"94!?C CCCC) c) ))))   #Y8($r2v,(h(48((((4$r(4(4tBx444445S>44d2h44444 4(4tCy4444'd2r6l'('t'''''d2h'h'4'''''uRW~'X'$'''''J'h'4''''T#57UVVVT#57UVVV # %6K  ' (r