s=open("probe.py").read()
s=s.replace('import tinker\nfrom common import *','import tinker, logging\nlogging.basicConfig(level=logging.INFO, format="%(asctime)s %(name)s %(message)s")\nfor n in ("httpx","httpcore","urllib3","filelock","huggingface_hub"): logging.getLogger(n).setLevel(logging.WARNING)\nfrom common import *')
a=s.index("    # restore into a new client"); b=s.index("    # one update on the probe client"); c=s.index('    res["ranks"][str(R)] = r')
restore=s[a:b].replace("tc2 = sc.create_training_client_from_state(st.path)","sc.close('success').result()\n    sc2 = service_client()\n    tc2 = sc2.create_training_client_from_state(st.path)")
restore+="    sc2.close('success').result()\n    sc = service_client()\n"
s=s[:a]+s[b:c]+restore+s[c:]
open("probe.py","w").write(s)