U *Ç-eV ã@sLddddddgZdd„Zdd„Zdd„Zd d „Zd d„Zd d„Zd d„ZdS)ÚhashableÚtransitive_getÚraisesÚ reverse_dictÚxfailÚfreezecCs*zt|ƒWdStk r$YdSXdS)NTF)ÚhashÚ TypeError)Úx©r úh/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/torch/fx/experimental/unification/utils.pyrs cCst|ƒr||kr||}q|S)zr Transitive dict.get >>> d = {1: 2, 2: 3, 3: 4} >>> d.get(1) 2 >>> transitive_get(1, d) 4 )r)ÚkeyÚdr r r r s cCs(z |ƒWdS|k r"YdSXdS)NFTr )ÚerrZlamdar r r rs cs¬t|ƒ‰dd„ˆ ¡Dƒ‰‡fdd„|Dƒ}g}|rŠ| ¡}| |¡| |d¡D]4}|ˆ|ksft‚ˆ| |¡ˆ|sR| |¡qRq0t‡fdd„|Dƒƒr¨t dƒ‚|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 cSsi|]\}}|t|ƒ“qSr )Úset)Ú.0ÚkÚvalr r r Ú 0sz_toposort..csh|]}|ˆkr|’qSr r ©rÚv©Zincoming_edgesr r Ú 1sz_toposort..r c3s|]}ˆ |d¡VqdS)N)Úgetrrr r Ú <sz_toposort..zInput has cycles) rÚitemsÚpopÚappendrÚAssertionErrorÚremoveÚaddÚanyÚ ValueError)ÚedgesÚSÚLÚnÚmr rr Ú _toposort!s r'cCs8i}|D]*}||D]}| |tƒ¡|f||<qq|S)a˜Reverses 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Útuple)r Úresultr rr r r rAs  cCs,z|ƒtdƒ‚Wntk r&YnXdS)NzXFailed test passed)Ú Exception)Úfuncr r r rRs  cCsTt|tƒrttt| ¡ƒƒSt|tƒr4ttt|ƒƒSt|ttfƒrPttt|ƒƒS|S)z¡ Freeze container to hashable form >>> freeze(1) 1 >>> freeze([1, 2]) (1, 2) >>> freeze({1: 2}) # doctest: +SKIP frozenset([(1, 2)]) ) Ú isinstanceÚdictÚ frozensetÚmaprrrr(Úlist)r r r r rZs  N)Ú__all__rrrr'rrrr r r r Ús