From 8eeba597b3b386f87388f9ad20c6276154dc3d2e Mon Sep 17 00:00:00 2001 From: Gahow Wang Date: Tue, 14 Jul 2026 12:52:00 +0800 Subject: [PATCH] Fix pilot burn-in selection manifest --- runs/fidelity-headroom/prepare_pilot.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/runs/fidelity-headroom/prepare_pilot.py b/runs/fidelity-headroom/prepare_pilot.py index bb849c4..168beb6 100644 --- a/runs/fidelity-headroom/prepare_pilot.py +++ b/runs/fidelity-headroom/prepare_pilot.py @@ -227,7 +227,11 @@ def build_manifest( for level, multiplier in TARGET_MULTIPLIERS.items(): target_rate = float(config["frontier_req_s_gpu"]) * multiplier target_count = round(target_rate * durations["low1"] * tp) - roles = [role for role in ROLES if role == "burnin" or role.startswith(level)] + roles = [ + role + for role in ROLES + if role.startswith(level) or (level == "low" and role == "burnin") + ] selections = {} for role in roles: anchor, selected = attainable_anchor(loaded[role][tier], target_count)