def METHOD_NAME(
    state: types.Array,
    action: types.Array,
    next_state: types.Array,
    discount: types.Array,
    reward: types.Array,
    step_type: types.Array,
    next_step_type: types.Array,
) -> trajectory.Transition:
  """Creates a Transition from current and next state information."""
  tfagents_time_step = ts.TimeStep(
      step_type=step_type,
      reward=np.zeros_like(reward),  # unknown
      discount=np.zeros_like(discount),  # unknown
      observation=state,
  )
  action_step = policy_step.PolicyStep(action=action, state=(), info=())
  tfagents_next_time_step = ts.TimeStep(
      step_type=next_step_type,
      reward=reward,
      discount=discount,
      observation=next_state,
  )
  return trajectory.Transition(
      time_step=tfagents_time_step,
      action_step=action_step,
      next_time_step=tfagents_next_time_step,
  )