L i+ZddlmZddlmZdgZdej dej fdZy))SequenceN set_tracegmreturncdttdttfd|jj fd5|j ddd|S#1swY|SxYw)a Sets a breakpoint in `gm`'s generated python code. It drops into pdb when `gm` gets run. Args: gm: graph module to insert breakpoint. It is then recompiled for it to take effect. Returns: the `gm` with breakpoint inserted. bodyrc dg|S)Nzimport pdb; pdb.set_trace() )rs a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torch/fx/experimental/debug.py insert_pdbzset_trace..insert_pdbs/7$77cfdS)Nc0r |S|S)Nr )r cur_transformr s r z-set_trace....s*M]4%8TtTr r )rr s`r rzset_trace..s Ur )make_transformerN)rstrlistgraphon_generate_code recompile)rr s @r rr se8#8498  " "  #    I Is AA$)collections.abcrtorch.fxfx__all__ GraphModulerr r r rs/$ -"..R^^r