DŽg ddlmZmZmZmZddlmZmZdeeefdedeeeffdZ deded eee eee ege fde fd Z y ) )AnyCallableDictOptional)ContextTreeSpec user_kwargsspecreturncV|jtusJ|jdk(sJ|jd}|jtusJt |t |j k7r$tdt|d|j i}|j D] }||||< |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. zkwarg key mismatch: Got z but expected ) typetuple num_childrenchildren_specsdictsetcontext ValueErrorlist)r r kwargs_specreordered_kwargskws `/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/export/_tree_utils.pyreorder_kwargsrs 99      !! !%%a(K   t ## # ;3{2233 $%^K4G4G3H J  !!/*2/ spec1spec2equivalence_fnc2||j|j|j|jsyt|jt|jk7ryt |j|jD]\}}t |||ryy)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( %**emmUZZ O 5   C(<(<$==$'(<(R>R$S [[+~F rN) typingrrrrtorch.utils._pytreerrstrrrboolr$rrr,s}001S#Xh4S>6  htngx~wOQUUV r