tasks/__init__.py

from .balanced_grid_easy import configure as balanced_grid_easy_configure, TASK_NAME as balanced_grid_easy_name
from .solar_management import configure as solar_management_configure, TASK_NAME as solar_management_name
from .wind_uncertainty import configure as wind_uncertainty_configure, TASK_NAME as wind_uncertainty_name
from .peak_demand import configure as peak_demand_configure, TASK_NAME as peak_demand_name
from .full_grid_challenge import configure as full_grid_challenge_configure, TASK_NAME as full_grid_challenge_name


TASKS = {
    balanced_grid_easy_name: balanced_grid_easy_configure,
    solar_management_name: solar_management_configure,
    wind_uncertainty_name: wind_uncertainty_configure,
    peak_demand_name: peak_demand_configure,
    full_grid_challenge_name: full_grid_challenge_configure,
}