tasks.py

def task_emergency_focus():
    """
    Task 1: Emergency Focus
    High number of emergencies, low normal patients. 
    Agent should learn to prioritize emergencies immediately.
    """
    return {
        "patients_waiting": 1,
        "emergency_patients": 4,
        "doctors_available": 2,
        "time_step": 0
    }

def task_balanced_load():
    """
    Task 2: Balanced Load
    Equal mix of normal and emergency patients.
    Agent needs to balance between priority and clearing the queue.
    """
    return {
        "patients_waiting": 3,
        "emergency_patients": 3,
        "doctors_available": 3,
        "time_step": 0
    }

def task_high_pressure():
    """
    Task 3: High Pressure
    Many patients, both normal and emergency, requiring efficient 
    actions to clear them before the 10 timestep limit.
    """
    return {
        "patients_waiting": 6,
        "emergency_patients": 5,
        "doctors_available": 2,
        "time_step": 0
    }