code: def clip_categorical_mpo_params(
      params: CategoricalMPOParams,
  ) -> CategoricalMPOParams:
      return params._replace(
          log_temperature=jnp.maximum(
              _MIN_LOG_TEMPERATURE, params.log_temperature
          ),
          log_alpha=jnp.maximum(_MIN_LOG_ALPHA, params.log_alpha),
      )
  
file_path: acme/acme/agents/jax/mpo/categorical_mpo.py
imports: from typing import NamedTuple, Tuple
  import distrax
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: clip_categorical_mpo_params(params) and returns: params