def get_stage_suffix(stage):
    """
    Which suffix to use (i.e. are we packaging a checkpoint?) - maps a stage to a suffix.
    :param stage:
    :return:
    """
    if stage == 'final':
        return ''
    else:
        return ".%s" % stage