code: def _insert_module_state(module: nn.Module, state: _State) -> None:
      global _module_state_mapping
      assert (
          module not in _module_state_mapping
      ), f"Inserting {module} more than once."
      _module_state_mapping[module] = state
  
imports: from typing import cast, Dict, Optional
  import torch.nn
label_desc: Use image node_id 1 for calling a global function with example usage: _insert_module_state(module, state) without return types