P/jhddlZddlmZmZddlZddlmZd dZdZdejjdeefdZ d de fd Z y) N)DictOptional) LazyStringc fddvrdd<dvr#tjjsdd<tfdS)z; Returns a LazyString that formats the graph code. cdSS)N )maybe_idnamesR/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/fx/_utils.py format_namez+lazy_format_graph_code..format_names  V1XJ' 'K print_outputFcoloredc tddjjjjdiS)Nz===== z ===== r )_format_graph_codeforward__code__ co_filenameprint_readable)r gmkwargssr z(lazy_format_graph_code..sA"[]O8 , JJ   + + B   ' ' r)sysstdoutisattyr)r rr rr s````@r lazy_format_graph_coder sO  V#!&~F3::#4#4#6!y   rcd|d|d|dS)z7 Returns a string that formats the graph code. zTRACED GRAPH r r )r filename graph_strs r rr#sTF!H:Qyk <}|j|j|jdk(rn|j}@dj |dddS)zD Returns a string of the last num_nodes nodes in the graph. rootrN)rangeappend format_noder(prevjoin)r* num_nodes node_contextscuris r get_node_contextr84siM C 9 S__./ 66V  hh  99]4R4( ))r)N))rtypingrrtorchtorch._loggingrrrfxGraphr+strr8r rr r@sF ! %4=uxx~~(4. *3 *r