code: def get_rlds_step_features() -> List[str]:
      """Returns a list representing features in an RLDS step."""
      return [
          rlds.OBSERVATION,
          rlds.ACTION,
          rlds.REWARD,
          rlds.DISCOUNT,
          rlds.IS_FIRST,
          rlds.IS_LAST,
          rlds.IS_TERMINAL,
      ]
  
file_path: agents/tf_agents/replay_buffers/rlds_to_reverb.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Callable, Dict, List, Optional, Tuple, Union
  from absl import logging
  import numpy
  import rlds
  import tensorflow
  from tf_agents.replay_buffers import reverb_utils
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.trajectories import trajectory
label_desc: Use image node_id 1 for calling a global function with example usage: get_rlds_step_features() and returns: list