&Vji ddlmZmZmZddlmZmZdeeefdedeeeffdZ deded eee eee ege fde fd Z d S) )AnyCallableOptional)ContextTreeSpec user_kwargsspecreturnc@|jtusJ|jdksJ|jd}|jtusJt |t |jkr'tdt|d|ji}|jD] }||||<|S)aXReorder user-provided kwargs to match the order in `spec`. `spec` is expected to be the in_spec of an exported program, i.e. the spec that results from flattening `(args, kwargs)`. We need this to provide consistent input ordering, such so that users can pass in foo(a=a, b=b) OR foo(b=b, a=a) and receive the same result. z>Ran into a kwarg keyword mismatch: Got the following keywords z but expected ) typetuple num_childrenchildren_specsdictsetcontext ValueErrorlist)rr kwargs_specreordered_kwargskws R/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/export/_tree_utils.pyreorder_kwargsrs 9       ! ! ! !%a(K  t # # # # ;3{23333 a*.{*;*; a aKVK^ a a   !//*2 spec1spec2equivalence_fnc||j|j|j|jsdSt|jt|jkrdSt |j|jD]\}}t |||sdSdS)a,Customizable equivalence check for two TreeSpecs. Arguments: spec1: The first TreeSpec to compare spec2: The second TreeSpec to compare equivalence_fn: A function to determine the equivalence of two TreeSpecs by examining their types and contexts. It will be called like: equivalence_fn(spec1.type, spec1.context, spec2.type, spec2.context) This function will be applied recursively to all children. Returns: True if the two TreeSpecs are equivalent, False otherwise. FT)rrlenrzip is_equivalent)rrr child_spec1 child_spec2s rr#r#!s( >%*emUZ O Ou 5   C(<$=$===u$'(R$S$S [[+~FF 55  4rN) typingrrrtorch.utils._pytreerrrstrrrboolr#rrr+s**********11111111S#Xh4S>6  htngx~wOQUUV r