Pulls out save_validation_info boolean logic

rafvasq · rafvasq · commit eed279a4c70e · 2025-11-18T15:18:44.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
@@ -812,7 +812,7 @@ def main():
     distributed_kwargs = get_distributed_kwargs(
         args.distributed, args.dist_timeout, args.save_validation_info_outputs
     )
-    save_validation_info_outputs = save_validation_info_outputs and (
+    args.save_validation_info_outputs = args.save_validation_info_outputs and (
         dist.get_rank() == 0
     )
 

Original file line number	Diff line number	Diff line change
`@@ -812,7 +812,7 @@ def main():`
`812`	`812`	`distributed_kwargs = get_distributed_kwargs(`
`813`	`813`	`args.distributed, args.dist_timeout, args.save_validation_info_outputs`
`814`	`814`	`)`
`815`		`- save_validation_info_outputs = save_validation_info_outputs and (`
	`815`	`+ args.save_validation_info_outputs = args.save_validation_info_outputs and (`
`816`	`816`	`dist.get_rank() == 0`
`817`	`817`	`)`
`818`	`818`