o h@shddlZddlmZddlZddlmZdddZddZdejj d ee fd d Z dd e fd dZ dS)N)Optional) LazyStringc slfdddvrdd<dvr,z tjsdd<Wn ty+dd<YnwtfddS)z; Returns a LazyString that formats the graph code. csdur dSS)N r)maybe_idnamerc/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/fx/_utils.py format_namesz+lazy_format_graph_code..format_name print_outputFcoloredcs(tddjjjjdiS)Nz===== z ===== r)_format_graph_codeforward__code__ co_filenameprint_readabler)r gmkwargsrrs  z(lazy_format_graph_code..)sysstdoutisattyAttributeErrorr)rrrrr)r rrrrrlazy_format_graph_code s   rcCsd|d|d|dS)z7 Returns a string that formats the graph code. zTRACED GRAPH r r)rfilename graph_strrrrr 'sr graphreturncCs2|jD]}|jdkrd|jvr|jdSqdS)zF Returns the nn_module_stack of the first call_function node. call_functionnn_module_stackN)nodesopmeta)rnoderrr#first_call_function_nn_module_stack.s r$cCsLg}|}t|D]}|||jdkrn|j}qd|dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappend format_noder!prevjoin)r# num_nodes node_contextscur_rrrget_node_context8s  r1)N)r%)rtypingrtorchtorch._loggingrrr fxGraphdictr$strr1rrrrs