tasks/balanced_grid_easy.py

"""
Task: balanced_grid_easy

Goal:
Balance renewable energy generation and demand with battery storage

Difficulty: Easy (solar generation is high and demand is moderate)
"""

TASK_NAME = "balanced_grid_easy"
BENCHMARK = "smart_grid"

def configure(env):
    env.config["base_demands"] = [15.0, 18.0, 12.0]
    
    env.config["solar_scale"] = 1.2
    env.config["wind_variation"] = "low"
    
    env.battery_capacity = 120
    env.battery_level = 60


def compute_score(total_reward: float, max_possible_reward: float) -> float:
    """
    Normalize all rewards into score in range [0, 1].
    """

    if max_possible_reward <= 0:
        return 0.0

    score = total_reward / max_possible_reward

    return min(max(score, 0.0), 1.0)