tjddlmZddlmZmZerddlmZmZddlm Z edZ gdZ d!d Z d"dZ d#dZd$dZd%dZd&dZd'dZd S)() annotations) TYPE_CHECKINGTypeVar)CallableIterable)Var_T)hashabletransitive_getraises reverse_dictxfailfreezexobjectreturnboolcH t|dS#t$rYdSwxYw)NTF)hash TypeError)rs e/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/fx/experimental/unification/utils.pyr r s9 Qt uus  !!keyddict[Var, object]cbt|r||vr||}t|r||v|S)zqTransitive dict.get >>> d = {1: 2, 2: 3, 3: 4} >>> d.get(1) 2 >>> transitive_get(1, d) 4 )r )rrs rr r s@ 3--C1HHf 3--C1HH Jerrtype[BaseException]lamdaCallable[[], object]c4 |dS#|$rYdSwxYw)NFT)rr s rr r &s9 u tts edgesdict[_T, Iterable[_T]]list[_T]ct|dDfd|D}g}|r|}||||dD]Z}||vrt d|d|d|||s||[|tfd|Drtd|S) a Topological sort algorithm by Kahn [1] - O(nodes + vertices) inputs: edges - a dict of the form {a: {b, c}} where b and c depend on a outputs: L - an ordered list of nodes that satisfy the dependencies of edges >>> # xdoctest: +SKIP >>> _toposort({1: (2, 3), 2: (3,)}) [1, 2, 3] Closely follows the wikipedia page [2] [1] Kahn, Arthur B. (1962), "Topological sorting of large networks", Communications of the ACM [2] http://en.wikipedia.org/wiki/Toposort#Algorithms c4i|]\}}|t|Sr#)set).0kvals r z_toposort..Bs$GGGfaaSGGGrch|]}|v| Sr#r#r*vincoming_edgess r z_toposort..Cs#555qQn44444rr#z Expected z in incoming_edges[]c3BK|]}|VdS)Ngetr/s r z_toposort..Os1 0 0Q>  a 0 0 0 0 0 0rzInput has cycles) ritemspopappendr6AssertionErrorremoveaddany ValueError)r$SLnmr1s @r _toposortrD3s="%((NGG0D0D0F0FGGGN5555E555A A  EEGG  1b!!  Aq)))$%K%K%Kq%K%K%KLLL 1  $ $Q ' ' '!!$ a  0 0 0 0% 0 0 000-+,,, Hrdict[_T, tuple[_T, ...]]cdi}|D]*}||D]}||d|fz||< +|S)aReverses direction of dependence dict. >>> d = {"a": (1, 2), "b": (2, 3), "c": ()} >>> reverse_dict(d) # doctest: +SKIP {1: ('a',), 2: ('a', 'b'), 3: ('b',)} .. note:: dict order are not deterministic. As we iterate on the input dict, it make the output of this function depend on the dict order. So this function output order should be considered as undeterministic. r#r5)rresultrr,s rrrTsWF77S6 7 7C **S"--6F3KK 7 MrfuncNonecX |td#t$rYdSwxYw)NzXFailed test passed) Exception)rHs rrrisC  -...      s  ))ct|tr4ttt|St|t r"ttt|St|ttfr"ttt|S|S)zFreeze container to hashable form >>> freeze(1) 1 >>> freeze([1, 2]) (1, 2) >>> freeze({1: 2}) # doctest: +SKIP frozenset([(1, 2)]) ) isinstancedict frozensetmaprr8r)tuplelist)rs rrrqs!T1VQWWYY//000!S)VQ(((!eT]##%S^^$$$ HrN)rrrr)rrrrrr)rrr r!rr)r$r%rr&)rr%rrE)rHr!rrI)rrrr) __future__rtypingrrcollections.abcrrvariabler r __all__r r r rDrrrr#rrrXs""""""))))))))22222222 WT]] V U U        B*          r