}YhzddlZddlmZddlZddlmZd dZdZdejj dee fdZ d de fd Z dS) N)Optional) LazyStringc fddvrdd<dvr: tjsdd<n#t$rdd<YnwxYwt fdS)z; Returns a LazyString that formats the graph code. cdSS)N )maybe_idnamesa/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/fx/_utils.py format_namez+lazy_format_graph_code..format_names"  ''X'' 'K print_outputFcoloredc rtddjjjjdiS)Nz===== z ===== r)_format_graph_codeforward__code__ co_filenameprint_readable)r gmkwargssr z(lazy_format_graph_code..sG" ,[[]] , , , J  + B  ' ' ' '  r )sysstdoutisattyAttributeErrorr)r rr rr s````@r lazy_format_graph_coder s  V##!&~F &:$$&& *$)y! & & & %F9    &         s#>AAcd|d|d|dS)z7 Returns a string that formats the graph code. zTRACED GRAPH r r)r filename graph_strs r rr's% =T < d|dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappend format_noder(prevjoin)r* num_nodes node_contextscur_s r get_node_contextr98s{M C 9  S__../// 6V   Eh 99]44R4( ) ))r )N)r,)rtypingrtorchtorch._loggingrrrfxGraphdictr+strr9rr r rAs  %%%%%%<===ux~(4. * *3 * * * * * *r