code: def identity_sample(
      output: networks_lib.NetworkOutput, key: types.PRNGKey
  ) -> networks_lib.Action:
      """Placeholder sampling function for non-distributional networks."""
      del key
      return output
  
file_path: acme/acme/agents/jax/bc/networks.py
imports: import dataclasses
  from typing import Callable, Optional
  from acme.jax import networks
  from acme.jax import types
  from typing_extensions import Protocol
label_desc: Use image node_id 1 for calling a global function with example usage: identity_sample(output, key) and returns: output