code: def _restore_io():
      _in_buf.seek(0)
      _out_buf.seek(0)
      _set_in_file(_in_buf)
      _set_out_file(_out_buf)
  
file_path: nni/test/ut/sdk/test_assessor.py
imports: from io import BytesIO
  import json
  from unittest import TestCase, main
  from nni.assessor import Assessor, AssessResult
  from nni.runtime import msg_dispatcher_base
  from nni.runtime.msg_dispatcher import MsgDispatcher
  from nni.runtime.tuner_command_channel.legacy import *
label_desc: Use image node_id 2 for calling a global function with example usage: _restore_io() without return types