code: def assert_unconsumed(key):
      """Assert that a key is unconsumed"""
      assert_consumed_value_p.bind(key, value=False)
  
file_path: jax/jax/experimental/key_reuse/_common.py
imports: from __future__ import annotations
  from typing import NamedTuple
  from jax import core
  from jax.interpreters import batching, mlir
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: assert_unconsumed(key) without return types