def get_dist_args(dist_type):
    """
    Get the distribution parameters that define the distribution, the first two numbers of each list define the current
    distribution and the last two define the target distribution.
    :param dist_type: string that indicates which distribution type
    :return: list of the different distribution parameters
    """
    if dist_type == 'uniform':
        dists = [[0, 1, 0, 1], [0, 1, 0.2, 1.2], [0, 1, 0.4, 1.4], [0, 1, 0.6, 1.6], [0, 1, 0.8, 1.8], [0, 1, 1, 2]]
    elif dist_type == 'uniform_wide':
        dists = [[0, 1, 0, 1], [0, 1, 0.0625, 1.1875], [0, 1, -0.1667, 1.5],
                 [0, 1, -0.375, 2.125], [0, 1, -1, 4], [0, 1, -4.25, 15.75]]
    elif dist_type == 'normal':
        dists = [[0, 1, 0, 1], [0, 1, 0.5, 1], [0, 1, 1.05, 1], [0, 1, 1.68, 1], [0, 1, 2.56, 1], [0, 1, 6, 1]]
    elif dist_type == 'normal2':
        dists = [[0, 1, 0, 1], [0, 1, 0, 1.52], [0, 1, 0, 2.41], [0, 1, 0, 4.25], [0, 1, 0, 10], [0, 1, 0, 100]]
    elif dist_type == 'multi-modal':  # currently also uniform, the shift is applied later
        dists = [[0, 1, 0, 1], [0, 1, 0, 1], [0, 1, 0, 1], [0, 1, 0, 1], [0, 1, 0, 1], [0, 1, 0, 1]]
    elif dist_type == 'categorical':
        category_1, category_2 = [0, 1, 2, 3], [0, 1, 2, 3]
        probs_1, probs_2, probs_3, probs_4, probs_5, probs_6 = [0.5, 0.5, 0.0, 0.0], [0.4, 0.4, 0.1, 0.1], \
                                                               [0.3, 0.3, 0.2, 0.2], [0.2, 0.2, 0.3, 0.3], \
                                                               [0.1, 0.1, 0.4, 0.4], [0.0, 0.0, 0.5, 0.5]
        dists = [[category_1, probs_1, category_2, probs_1], [category_1, probs_1, category_2, probs_2],
                 [category_1, probs_1, category_2, probs_3], [category_1, probs_1, category_2, probs_4],
                 [category_1, probs_1, category_2, probs_5], [category_1, probs_1, category_2, probs_6]]
    else:
        print('invalid dist type choose: uniform, normal, multi-modal, categorical')
        dists = None
    return dists