"""Environment registry for NephroSim benchmarks."""

BENCHMARK_ENVS = [
    {
        "env_id": "nephrosim/UreaClearing-v0",
        "difficulty": "easy",
        "category": "urea-clearance",
        "description": "Optimize Qb and Qd for maximum Kt/V",
    },
    {
        "env_id": "nephrosim/UltrafiltrationControl-v0",
        "difficulty": "easy",
        "category": "hemodynamic-stability",
        "description": "Control UF rate to prevent intradialytic hypotension",
    },
    {
        "env_id": "nephrosim/PhosphateMgmt-v0",
        "difficulty": "easy",
        "category": "phosphate-homeostasis",
        "description": "Optimize phosphate binder dosing across weekly cycle",
    },
    {
        "env_id": "nephrosim/FullDialysisSession-v0",
        "difficulty": "hard",
        "category": "multi-objective",
        "description": "Joint optimization of clearance and hemodynamics",
    },
]
