code: def _get_discount(experience) -> types.Tensor:
      """Try to get the discount entry from `experience`.
  
      Typically experience is either a Trajectory or a Transition.
  
      Args:
        experience: Data collected from e.g. a replay buffer.
  
      Returns:
        discount: The discount tensor stored in `experience`.
      """
      if isinstance(experience, trajectory.Transition):
          return experience.time_step.discount
      else:
          return experience.discount
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from typing import Optional, Text, Tuple
  from absl import logging
  import gin
  from six.moves import range
  from six.moves import zip
  import tensorflow
  from tf_agents.agents import data_converter
  from tf_agents.agents import tf_agent
  from tf_agents.agents.ppo import ppo_policy
  from tf_agents.agents.ppo import ppo_utils
  from tf_agents.networks import network
  from tf_agents.policies import greedy_policy
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.trajectories import trajectory
  from tf_agents.typing import types
  from tf_agents.utils import common
  from tf_agents.utils import eager_utils
  from tf_agents.utils import nest_utils
  from tf_agents.utils import object_identity
  from tf_agents.utils import tensor_normalizer
  from tf_agents.utils import value_ops
label_desc: Use image node_id 2 for calling a global function with example usage: _get_discount(experience) and returns: experience, experience