code: def set_log_probability(
      info: types.NestedTensorOrArray, log_probability: types.Float
  ) -> types.NestedTensorOrArray:
      """Sets the CommonFields.LOG_PROBABILITY on info to be log_probability."""
      if info in ((), None):
          return PolicyInfo(log_probability=log_probability)
      return _maybe_set_value_namedtuple_or_dict(
          info, CommonFields.LOG_PROBABILITY, log_probability
      )
  
file_path: agents/tf_agents/trajectories/policy_step.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from typing import Any, Mapping, NamedTuple, Optional, Text, Union
  from tf_agents.typing import types
label_desc: Use image node_id 3 for calling a global function with example usage: set_log_probability(info, log_probability) and returns: _maybe_set_value_namedtuple_or_dict, PolicyInfo