code: def get_default_behavior_policy(
      networks: TD3Networks,
      action_specs: specs.BoundedArray,
      sigma: float,
  ) -> actor_core_lib.FeedForwardPolicy:
      """Selects action according to the policy."""
  
      def behavior_policy(
          params: networks_lib.Params,
          key: networks_lib.PRNGKey,
          observation: types.NestedArray,
      ):
          action = networks.policy_network.apply(params, observation)
          noise = jax.random.normal(key, shape=action.shape) * sigma
          noisy_action = jnp.clip(
              action + noise, action_specs.minimum, action_specs.maximum
          )
          return noisy_action
  
      return behavior_policy
  
file_path: acme/acme/agents/jax/td3/networks.py
imports: import dataclasses
  from typing import Callable, Sequence
  from acme import specs
  from acme import types
  from acme.agents.jax import actor_core
  from acme.jax import networks
  from acme.jax import utils
  import haiku
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: get_default_behavior_policy(networks, action_specs, sigma) and returns: behavior_policy, noisy_action