Add missing used_keys addition (#189)

rafvasq · web-flow · commit 57fcb66ac9c2 · 2026-02-13T15:32:03.000-05:00
Signed-off-by: Rafael Vasquez &lt;rafvasq21@gmail.com&gt;
diff --git a/aiu_fms_testing_utils/scripts/drive_paged_programs.py b/aiu_fms_testing_utils/scripts/drive_paged_programs.py
@@ -871,6 +871,7 @@ def get_valid_prompts(
                                 allow_truncation=allow_truncation,
                                 enforce_sizes=enforce_sizes,
                             )
+                            used_keys.add(program_seq_key[0])
                             yield ValidPrompt(
                                 program_id=program_seq_key[0],
                                 shape=valid_prompt_shape,

Original file line number	Diff line number	Diff line change
`@@ -871,6 +871,7 @@ def get_valid_prompts(`
`871`	`871`	`allow_truncation=allow_truncation,`
`872`	`872`	`enforce_sizes=enforce_sizes,`
`873`	`873`	`)`
	`874`	`+ used_keys.add(program_seq_key[0])`
`874`	`875`	`yield ValidPrompt(`
`875`	`876`	`program_id=program_seq_key[0],`
`876`	`877`	`shape=valid_prompt_shape,`