def state_size_dataset(sz):
    """Get dataset key part for state size.

    Parameters
    ----------
    sz : `int`
        State size.

    Returns
    -------
    `str`
        Dataset key part.
    """
    return '_ss%d' % sz