code: def init_state():
      state = State()
      STATE[threading.get_ident()] = state
      return state
  
file_path: autokeras/autokeras/prototype/graph_state.py
imports: import threading
  from tensorflow import keras
label_desc: Use image node_id 2 for calling a global function with example usage: init_state() and returns: state