tasks/full_grid_challenge.py

"""
Task: full_grid_challenge

Goal:
Manage high variable supply and demand with limited battery capacity

Difficulty: Very Hard (unpredictable wind, noisy demand & limited storage)
"""

from .utils import compute_score

TASK_NAME = "full_grid_challenge"
BENCHMARK = "smart_grid"

def configure(env):
    env.config["base_demands"] = [28.0, 35.0, 30.0]
    env.config["demand_noise"] = "high"
    env.config["peak_demand_multiplier"] = 1.4
    
    env.config["solar_scale"] = 1.0
    env.config["wind_variation"] = "high"
    
    env.battery_capacity = 70
    env.battery_level = 50