code: def policy_loss_coeff_advantage_indicator(
      networks: CRRNetworks,
      policy_params: networks_lib.Params,
      critic_params: networks_lib.Params,
      transition: types.Transition,
      key: networks_lib.PRNGKey,
      num_action_samples: int = 4,
  ) -> jnp.ndarray:
      """Indicator advantage weighting; see equation (3) in CRR paper."""
      advantage = _compute_advantage(
          networks,
          policy_params,
          critic_params,
          transition,
          key,
          num_action_samples,
      )
      return jnp.heaviside(advantage, 0.0)
  
file_path: acme/acme/agents/jax/crr/losses.py
imports: from typing import Callable
  from acme import types
  from acme.agents.jax.crr.networks import CRRNetworks
  from acme.jax import networks
  import jax.numpy
label_desc: Use image node_id 3 for calling a global function with example usage: policy_loss_coeff_advantage_indicator(networks, policy_params, critic_params, transition, key, num_action_samples) and returns: jnp