_conditions = {
    0: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    1: {
        "mu_hidden": [64, 64],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    2: {
        "mu_hidden": [256, 256],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    3: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    4: {
        "mu_hidden": [1024, 1024],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    5: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    6: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    7: {
        "mu_hidden": [64, 64],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    8: {
        "mu_hidden": [256, 256],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    9: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    10: {
        "mu_hidden": [1024, 1024],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    11: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    12: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    13: {
        "mu_hidden": [64, 64],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    14: {
        "mu_hidden": [256, 256],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    15: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    16: {
        "mu_hidden": [1024, 1024],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    17: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    18: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    19: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [64, 64],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    20: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [256, 256],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    21: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    22: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [1024, 1024],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    23: {
        "mu_hidden": [32, 32],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    24: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    25: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [64, 64],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    26: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [256, 256],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    27: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    28: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [1024, 1024],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    29: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    30: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    31: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [64, 64],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
    32: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [256, 256],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    33: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [512, 512],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    34: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [1024, 1024],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    35: {
        "mu_hidden": [2048, 2048],
        "sigma_hidden": [2048, 2048],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 512,
    },
    36: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 1, "target": 2.5},
        "batch_size": 2048,
    },
    37: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 10, "target": 2.5},
        "batch_size": 2048,
    },
    38: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 50, "target": 2.5},
        "batch_size": 2048,
    },
    39: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 90, "target": 2.5},
        "batch_size": 2048,
    },
    40: {
        "mu_hidden": [512, 512],
        "sigma_hidden": [32, 32],
        "velocity_ratio_params": {"enforce": 100, "target": 2.5},
        "batch_size": 2048,
    },
}

def get_conditions(n_seeds: int = 5) -> dict:
    conditions = []
    for seed in range(n_seeds):
        for en, condition in _conditions.items():
            condition['seed'] = seed
            condition['condition'] = en
            conditions.append(condition.copy())
    return conditions
