code: def reset_thread_local_state_for_test():
      global _thread_local_state
      _thread_local_state = _ThreadState()
  
imports: import collections
  import contextlib
  import threading
  from typing import TypeVar, Optional, Union
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import module
  import jmp
label_desc: Use image node_id 2 for calling a global function with example usage: reset_thread_local_state_for_test() without return types