def METHOD_NAME(shell_type):
    '''
    Helper function to adjust the ordering of the internal AO in each shell if needed.
    Parameters
    ----------
        shell_type: int, 0 for s, 1 for p, 2 for d.
    Returns
    ----------
        a dictionary of how each AO should be shifted compared to their original location.
    '''
    if abs(shell_type) == 0:
        return {0: 0}
    elif abs(shell_type) == 1:
        return {0: 0, 1: 0, 2: 0}
    elif abs(shell_type) == 2:
        return {0: 0, 1: 0, 2: 0, 3: 0, 4: 0, 5: 0}
    else:
        raise KeyError("only s, p, d shell is allowed.")