Function main

codeclash/utils/generate_confs_multi.py:10–42

(models: str, arenas: str, rounds: int, simulations: int, record_ratio: float, output: Path)

def main(models: str, arenas: str, rounds: int, simulations: int, record_ratio: float, output: Path):
    # Get all models
    models = yaml.safe_load(open(models))
    output.mkdir(parents=True, exist_ok=True)

    # Get arenas
    arenas_list = ARENAS if arenas == "all" else [a for a in ARENAS if a.name in arenas.split(",")]
    if not arenas_list:
        print(f"No valid arenas found from {arenas}. Choose from {[a.name for a in ARENAS]}.")
        return  # Stop execution if no valid arenas are found

    for arena in arenas_list:
        print(f"Generating config for arena: {arena.name}")
        config = get_config(rounds, simulations, arena, models)
        config_name = f"{arena.name}__p{len(models)}__r{rounds}__s{simulations}.yaml"
        with open(output / config_name, "w") as f:
            yaml.dump(
                config,
                f,
                default_style=None,
                sort_keys=False,
                allow_unicode=True,
                default_flow_style=False,
                Dumper=yaml.SafeDumper,
            )

        clean_config(output / config_name)

    print(f"Generated {len(arenas_list)} configuration files in '{output}'.")
    print(f"- # Models: {len(models)}")
    print(f"- # Arenas: {len(arenas_list)}")
    print(f"- r (rounds) {rounds}")
    print(f"- s (sims_per_round) {simulations}")


if __name__ == "__main__":