def get_rollout_size(rollout):
    """Gets a rollout's size.

    Args:
      rollout: a rollout (dict, a return from `sample_rollout` func).

    Returns:
      length of the rollout (int)
    """
    return len(rollout['reward'])
