From e152a91c05c217db904f979624fe0d8012894be1 Mon Sep 17 00:00:00 2001 From: Nils Date: Tue, 14 Jul 2026 12:50:45 +0200 Subject: [PATCH] parse_known_args: module-level parser must tolerate importers' argv Co-Authored-By: Claude Fable 5 --- scripts/train_merge_code.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train_merge_code.py b/scripts/train_merge_code.py index 00fb7a5..8f8acfe 100644 --- a/scripts/train_merge_code.py +++ b/scripts/train_merge_code.py @@ -51,7 +51,7 @@ ap.add_argument("--deepk", type=int, default=0, ap.add_argument("--bptt", type=int, default=0, help="truncated BPTT: grads only through last N iterations") ap.add_argument("--lr", type=float, default=1e-3) -ARGS = ap.parse_args() +ARGS, _ = ap.parse_known_args() SEED = ARGS.seed LR = ARGS.lr SUFFIX = ((f"_s{SEED}" if SEED else "")