def get_weight(mode: Mode, count: int):
    if (count == 1):
        return [1.0]
    else:
        return {Mode.EQUAL: equal(count), Mode.GAUSS: gauss(count), Mode.GAUSS_SYM: gauss_sym(count), Mode.PYRAMID: pyramid(count), Mode.PYRAMID_SYM: pyramid_sym(count), Mode.SIVEROO_1: siveroo1(count), Mode.SIVEROO_2: stretch(count, [1, 3, 3, 2, 2])}.get(mode, [1, 0])
def get_weight(mode: Mode, count: int):
    if (count == 1):
        return [1.0]
    else:
        return {Mode.EQUAL: equal(count), Mode.GAUSS: gauss(count), Mode.GAUSS_SYM: gauss_sym(count), Mode.PYRAMID: pyramid(count), Mode.PYRAMID_SYM: pyramid_sym(count), Mode.SIVEROO_1: siveroo1(count), Mode.SIVEROO_2: stretch(count, [1, 3, 3, 2, 2])}.get(mode, [1, 0])