pub struct TaskState {
    pub previous_energy_usage: u64,
    pub budget: u64,
    pub children: Vec<u64>,
}