o ho @sUdZddlZddlZddlZddlmZddlmZmZddl Z iZ e e efed<eZee ed<dee defd d Zd e ddfd d Zdeejdee e effddZdedee e effddZ ddededee e effddZdS)zG Utilities for converting data types into structured JSON for dumping. N)Sequence)AnyOptional INTERN_TABLE DUMPED_FILESsreturncsRdurdStddur'ttt<tjjjdfddddS)NstrcsfSNr rrr m/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_logging/structured.pyszintern_string..T)suppress_context)rgetlentorch_logging _internaltrace_structured)rr r r intern_strings rfilenamecsXdvrdStvr dStddlmtjjjdfddfddddS) N eval_with_keyr)_loader dump_filecsdiS)Nnamer r rr rr+szdump_file..cs Sr ) get_sourcer rrr rr.s ) metadata_fn payload_fn)raddtorch.fx.graph_modulerrrrrrr r rr!s     rtbcCsdd|D}|S)NcSs&g|]}|j|jt|j|jdqS))linerrloc)linenorrrr&.0framer r r 5sz"from_traceback..r )r%rr r rfrom_traceback2s r- num_framesc Csddlm}ddlm}|}|rt|d|dS|}tj t t tj j}tt|D]\}}||jvrT|t|||t||}t|Sq5t|d|dS)Nr)TracingContextCapturedTracebackr ) torch._guardsr/torch.utils._tracebackr1 extract_stackr-extractsummaryospathdirnameinspectgetfilersep enumeratereversedrr) r.r/r1user_tbr%torch_filepathir+ filtered_tbr r rget_user_stackAs      rCFcppcsNddlmddlm}|j|d}fdd|D}t|d|dS) zJ Returns the traceback for the user stack and the framework stack runinteresting_filesr0)rEcs<g|]}|jdr|jvsd|jvsd|jvr|qS)z.pyzat::ztorch::)rendswithrr)rFr rr,`s  z'get_framework_stack..r N)%torch.fx.experimental.symbolic_shapesrGr3r1r5r6r-)r.rEr1r%r rFrget_framework_stackVs    rJ)rDF)__doc__r:r7 tracebackcollections.abcrtypingrrtorch._logging._internalrrdictr int__annotations__setrrr FrameSummarylistr-rCboolrJr r r rs* $