code: def categorical_importance_sampling_ratios(
      pi_logits_t: Array, mu_logits_t: Array, a_t: Array
  ) -> Array:
      """Compute importance sampling ratios from logits.
  
      Args:
        pi_logits_t: unnormalized logits at time t for the target policy.
        mu_logits_t: unnormalized logits at time t for the behavior policy.
        a_t: actions at time t.
  
      Returns:
        importance sampling ratios.
      """
      warnings.warn(
          "Rlax categorical_importance_sampling_ratios will be deprecated. "
          "Please use distrax.importance_sampling_ratios instead.",
          PendingDeprecationWarning,
          stacklevel=2,
      )
      return distrax.importance_sampling_ratios(
          distrax.Categorical(pi_logits_t),
          distrax.Categorical(mu_logits_t),
          a_t,
      )
  
file_path: rlax/rlax/_src/distributions.py
imports: import collections
  import warnings
  import chex
  import distrax
  import jax.numpy
label_desc: Use image node_id 8 for calling a global function with example usage: categorical_importance_sampling_ratios(pi_logits_t, mu_logits_t, a_t) and returns: distrax