Minor fix for getting function's graph#2504

gramalingam merged 2 commits intomainmicrosoft/onnxscript:mainfrom

rama/fixmicrosoft/onnxscript:rama/fixCopy head branch name to clipboard