code: def _is_mutable(axis_col: str) -> bool:
      """Determines whether a collection is mutable.
  
      For example, when a module is called with `module.apply(..., mutable=['z'])`,
      this function will return True for `axis_col='z'` and False otherwise.
  
      If there is no module in scope, this function will return True.
  
      Args:
        axis_col: Name of the collection in question.
  
      Returns:
        Whether it is currently mutable.
      """
      last = nn.module._context.module_stack[
          -1
      ]  # pylint: disable=protected-access
      if last:
          return last.is_mutable_collection(axis_col)
      else:
          return True
  
imports: import functools
  import re
  from typing import Any, Callable, Mapping, Optional, Tuple
  import flax
  from flax import linen
  from flax import struct
  from flax.core.frozen_dict import freeze
  from flax.core.frozen_dict import unfreeze
  from flax.core.lift import In
  from flax.core.lift import Out
  from flax.linen.spmd import _axis_rules
  from flax.linen.spmd import _AxisRules
  from flax.linen.spmd import _is_logical_spec
  from flax.linen.spmd import _with_sharding_constraint
  from flax.linen.spmd import Array
  from flax.linen.spmd import ArrayPytree
  from flax.linen.spmd import get_logical_axis_rules
  from flax.linen.spmd import logical_axis_rules
  from flax.linen.spmd import logical_to_mesh
  from flax.linen.spmd import logical_to_mesh_axes
  from flax.linen.spmd import LogicalPartitionSpec
  from flax.linen.spmd import LogicalPartitionSpecPytree
  from flax.linen.spmd import LogicalRules
  from flax.linen.spmd import PartitionSpecPytree
  from flax.linen.spmd import RulesFallback
  from flax.linen.spmd import set_logical_axis_rules
  from flax.linen.spmd import with_logical_constraint
  from flax.traverse_util import flatten_dict
  from flax.traverse_util import unflatten_dict
  import jax
label_desc: Use image node_id 7 for calling a global function with example usage: _is_mutable(axis_col) and returns: last, True