Search commited on
Commit
607c4cf
·
1 Parent(s): 8769d1f

auto: sync run_qwen_anchor_geometry_profile_sweep.py

Browse files
scripts/run_qwen_anchor_geometry_profile_sweep.py CHANGED
@@ -245,29 +245,39 @@ def build_markdown_report(
245
  return "\n".join(lines)
246
 
247
 
248
- def main() -> None:
249
  parser = argparse.ArgumentParser(description="Sweep anchor span profiles for geometry-only crystallization analysis.")
250
- parser.add_argument("--model", type=str, default="Qwen/Qwen3.5-4B")
251
  parser.add_argument("--device", type=str, default="cpu")
252
- parser.add_argument("--max_length", type=int, default=160)
253
  parser.add_argument(
254
  "--profiles",
 
 
255
  nargs="+",
256
  default=list(list_anchor_span_profiles()),
257
  )
258
- parser.add_argument("--case_name", type=str, default=None)
259
  parser.add_argument("--limit", type=int, default=None)
260
  parser.add_argument(
261
  "--output_json",
 
 
262
  type=Path,
263
  default=ROOT / "archive" / "qwen_anchor_geometry_profile_sweep.json",
264
  )
265
  parser.add_argument(
266
  "--output_md",
 
 
267
  type=Path,
268
  default=ROOT / "docs" / "research" / "qwen_anchor_geometry_profile_sweep.md",
269
  )
270
- args = parser.parse_args()
 
 
 
 
271
 
272
  for profile in args.profiles:
273
  if profile not in list_anchor_span_profiles():
 
245
  return "\n".join(lines)
246
 
247
 
248
+ def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
249
  parser = argparse.ArgumentParser(description="Sweep anchor span profiles for geometry-only crystallization analysis.")
250
+ parser.add_argument("--model", "--model-name", dest="model", type=str, default="Qwen/Qwen3.5-4B")
251
  parser.add_argument("--device", type=str, default="cpu")
252
+ parser.add_argument("--max_length", "--max-length", dest="max_length", type=int, default=160)
253
  parser.add_argument(
254
  "--profiles",
255
+ "--anchor-profile",
256
+ dest="profiles",
257
  nargs="+",
258
  default=list(list_anchor_span_profiles()),
259
  )
260
+ parser.add_argument("--case_name", "--case-name", dest="case_name", type=str, default=None)
261
  parser.add_argument("--limit", type=int, default=None)
262
  parser.add_argument(
263
  "--output_json",
264
+ "--output-json",
265
+ dest="output_json",
266
  type=Path,
267
  default=ROOT / "archive" / "qwen_anchor_geometry_profile_sweep.json",
268
  )
269
  parser.add_argument(
270
  "--output_md",
271
+ "--output-md",
272
+ dest="output_md",
273
  type=Path,
274
  default=ROOT / "docs" / "research" / "qwen_anchor_geometry_profile_sweep.md",
275
  )
276
+ return parser.parse_args(argv)
277
+
278
+
279
+ def main() -> None:
280
+ args = parse_args()
281
 
282
  for profile in args.profiles:
283
  if profile not in list_anchor_span_profiles():