h ^UddlmZddlmZmZmZmZddlmZddl Z ddl m Z m Z mZee egefZee egefZiZeeeefed<iZeeeeefed<edZed Zed Z dd eed ed eeddfdZd eeddfdZde dedeefdZdeeefdedeefdZdeededeefdZde edfdedeefdZ!dededeefdZ"deeefdedefdZ#deededefdZ$de edfdedefdZ%dededefdZ&eeee#eeee$ee e!e%e j'j(D]Z)ee)e!e%eee"e&dS)) namedtuple)AnyCallableOptionalTypeVar) NamedTupleN)PyTree tree_flattenTreeSpecSUPPORTED_NODESSUPPORTED_NODES_EXACT_MATCH_T_K_Vclsflatten_fn_specflatten_fn_exact_match_specreturnc.|t|<|t|<dSNr r )rrrs X/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/torch/fx/_pytree.pyregister_pytree_flatten_specrs +OC'B$$$c&t|=t|=dSrr)rs r_deregister_pytree_flatten_specrs #C(((rpytreespecc V|r|gS|jtvrRt|j}|||}g}t||jD]\}}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 zipchildren_specstree_flatten_specr RuntimeError) rrr child_pytreesresultchild child_specflat flat_result real_specs rr$r$$s ||~~x  yO##)$)4'55 !$]D4G!H!H   E:$UJ77D dNFF )&11K y m m mv m mX\ m m m    rdc*fd|jDS)Nc g|] }| Sr0).0kr-s r z&_dict_flatten_spec..As ' ' 'QAaD ' ' 'r)contextr-rs` r_dict_flatten_specr6@s ' ' ' '$, ' ' ''rcDfdt|jDS)Nc g|] }| Sr0r0r1ir-s rr3z&_list_flatten_spec..E 3 3 3QAaD 3 3 3rrange num_childrenr5s` r_list_flatten_specr?D( 3 3 3 3% 122 3 3 33r.cDfdt|jDS)Nc g|] }| Sr0r0r9s rr3z'_tuple_flatten_spec..Ir;rr<r5s` r_tuple_flatten_specrCHr@rcDfdt|jDS)Nc g|] }| Sr0r0r9s rr3z,_namedtuple_flatten_spec..Mr;rr<r5s` r_namedtuple_flatten_specrFLr@rc2t||jkSrlenr>r5s r_dict_flatten_spec_exact_matchrJP q66T& &&rc2t||jkSrrHr5s r_list_flatten_spec_exact_matchrMTrKrc2t||jkSrrHr5s r_tuple_flatten_spec_exact_matchrOXrKrc2t||jkSrrHr5s r$_namedtuple_flatten_spec_exact_matchrQ\rKrr)* collectionsrtypingrrrrtyping_extensionsrtorch.return_typestorchtorch.utils._pytreer r r listFlattenFuncSpecboolFlattenFuncExactMatchSpecr dictr!__annotations__r rrrrrr$r6r?tuplerCrFrJrMrOrQ return_typesall_return_types return_typer0rrrbs"""""""333333333333((((((>>>>>>>>>>FH-t34$fh%7%=>46d3i01666TVT$s)X6O-P"PQVVV WT]] WT]] WT]] HLCC cC$C"**C!DC 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