code: def reserve_up_to_full_rng_block():
      """If RNG is active in the current frame, reserve up to the default block."""
      # TODO(lenamartens): Fix needing full block reservation in stateful
      # control-flow by keeping track of current key with index and keeping a full
      # block in PRNGSequence at all time.
      rng_seq = base.current_frame().rng_stack.peek()
      if rng_seq:
          rng_seq.reserve_up_to_full()
  
imports: import collections
  import collections.abc
  from collections.abc import Mapping, MutableMapping
  import functools
  import inspect
  from typing import Any, Callable, Optional, TypeVar
  from haiku._src import base
  import jax
  import jax.numpy
label_desc: Use image node_id 6 for calling a global function with example usage: reserve_up_to_full_rng_block() without return types