Sidharth1743 commited on
Commit
ee52ecc
·
1 Parent(s): c8d80d5

task 3 and task 4 refined

Browse files
Files changed (3) hide show
  1. inference.py +1 -1
  2. server/graders.py +1 -1
  3. server/tasks.py +2 -2
inference.py CHANGED
@@ -705,7 +705,7 @@ def choose_action_with_qwen(
705
  },
706
  }
707
 
708
- if task_id in {"single_fault", "n_minus_1", "cascade_prevent"}:
709
  selected_outcome = choose_best_simulation(
710
  task_id=task_id,
711
  observation=observation,
 
705
  },
706
  }
707
 
708
+ if task_id in {"single_fault", "n_minus_1"}:
709
  selected_outcome = choose_best_simulation(
710
  task_id=task_id,
711
  observation=observation,
server/graders.py CHANGED
@@ -126,7 +126,7 @@ def grade_cascade_prevent(
126
 
127
 
128
  def grade_multi_stage_cascade(
129
- episode_log: list[EpisodeStepLog], max_steps: int = 25
130
  ) -> float:
131
  if not episode_log:
132
  return 0.01
 
126
 
127
 
128
  def grade_multi_stage_cascade(
129
+ episode_log: list[EpisodeStepLog], max_steps: int = 30
130
  ) -> float:
131
  if not episode_log:
132
  return 0.01
server/tasks.py CHANGED
@@ -55,10 +55,10 @@ TASKS: Dict[TaskId, TaskSpec] = {
55
  difficulty="hard",
56
  description=(
57
  "Three lines are disconnected and load is increased by 20%. Manage the "
58
- "guaranteed three-stage cascade for 25 steps and preserve as much load "
59
  "as possible across stage boundaries."
60
  ),
61
- max_steps=25,
62
  ),
63
  }
64
 
 
55
  difficulty="hard",
56
  description=(
57
  "Three lines are disconnected and load is increased by 20%. Manage the "
58
+ "guaranteed three-stage cascade for 30 steps and preserve as much load "
59
  "as possible across stage boundaries."
60
  ),
61
+ max_steps=30,
62
  ),
63
  }
64