def METHOD_NAME(
    transitions: types.Transition) -> reverb.ReplaySample:
  """Converts a types.Transition to a reverb.ReplaySample."""
  info = tree.map_structure(lambda dtype: tf.ones([], dtype),
                            reverb.SampleInfo.tf_dtypes())
  return reverb.ReplaySample(info=info, data=transitions)