Allow named dynamic exit search outputs

This commit is contained in:
Codex
2026-06-28 06:52:52 +08:00
parent 340e220b28
commit 5a9786d861
3 changed files with 6 additions and 1 deletions
@@ -30,6 +30,7 @@ def main() -> None:
parser.add_argument("--second-target-multipliers", type=_float_tuple)
parser.add_argument("--take1-ratios", type=_float_tuple)
parser.add_argument("--take2-ratios", type=_float_tuple)
parser.add_argument("--output-dir-name", default="dynamic-exit-search")
args = parser.parse_args()
setup_logging()
search_dynamic_exit_plans(args)
+1
View File
@@ -108,6 +108,7 @@ class TrainingContractTest(unittest.TestCase):
second_target_multipliers=(2.0,),
take1_ratios=(0.5,),
take2_ratios=(0.25,),
output_dir_name="dynamic-exit-search",
)
)
@@ -123,7 +123,10 @@ def search_dynamic_exit_plans(args: Any) -> None:
"robust_candidate_found": bool(best["robust_candidate_found"]),
"best_plan": best,
}
out_dir = root / "dynamic-exit-search"
output_dir_name = str(getattr(args, "output_dir_name", None) or "dynamic-exit-search")
if output_dir_name in {"", ".", ".."} or "/" in output_dir_name or "\\" in output_dir_name:
raise ValueError(f"output_dir_name must be a run-local directory name: {output_dir_name}")
out_dir = root / output_dir_name
write_json(out_dir / "dynamic_exit_search_result.json", _jsonable(payload))
write_text(out_dir / "dynamic_exit_search_rows.csv", result.to_csv(index=False))
write_text(out_dir / "dynamic_exit_search_summary.csv", summary.to_csv(index=False))