Claude commited on
Commit
03d9529
·
unverified ·
1 Parent(s): 97b6de5

Add --llm-agent and other legacy CLI flags for backwards compatibility

Browse files

External runner passes --llm-agent which was dropped in the config.yaml
refactor. Re-add it (always true, deprecated) plus --report-dir,
--log-dir, --eval-episodes, --example-customers overrides.

https://claude.ai/code/session_01DPirJ78YYN4fJUvUFJ5D6V

Files changed (1) hide show
  1. layer1/train.py +21 -0
layer1/train.py CHANGED
@@ -170,6 +170,19 @@ def main():
170
  help="Prompt to evaluate (eval mode)")
171
  parser.add_argument("--no-report", action="store_true",
172
  help="Skip report generation")
 
 
 
 
 
 
 
 
 
 
 
 
 
173
  args = parser.parse_args()
174
 
175
  # Load config from YAML
@@ -188,6 +201,14 @@ def main():
188
  paths_cfg["output_dir"] = args.output_dir
189
  if args.no_report:
190
  report_cfg["enabled"] = False
 
 
 
 
 
 
 
 
191
 
192
  if args.mode == "train":
193
  run_train(grpo_config, report_cfg, paths_cfg, args.hf_token)
 
170
  help="Prompt to evaluate (eval mode)")
171
  parser.add_argument("--no-report", action="store_true",
172
  help="Skip report generation")
173
+ parser.add_argument("--report-dir", type=str, default=None,
174
+ help="Override report output directory from config")
175
+ parser.add_argument("--log-dir", type=str, default=None,
176
+ help="Override log directory from config")
177
+ parser.add_argument("--eval-episodes", type=int, default=None,
178
+ help="Override eval episodes for report from config")
179
+ parser.add_argument("--example-customers", type=int, default=None,
180
+ help="Override example customers in report from config")
181
+ parser.add_argument("--output", type=str, default=None,
182
+ help="Save results to JSON file")
183
+ # Legacy flags (accepted for backwards compatibility with external runners)
184
+ parser.add_argument("--llm-agent", action="store_true", default=True,
185
+ help="(deprecated, always true) Use LLM agent")
186
  args = parser.parse_args()
187
 
188
  # Load config from YAML
 
201
  paths_cfg["output_dir"] = args.output_dir
202
  if args.no_report:
203
  report_cfg["enabled"] = False
204
+ if args.report_dir is not None:
205
+ report_cfg["output_dir"] = args.report_dir
206
+ if args.log_dir is not None:
207
+ paths_cfg["log_dir"] = args.log_dir
208
+ if args.eval_episodes is not None:
209
+ report_cfg["eval_episodes"] = args.eval_episodes
210
+ if args.example_customers is not None:
211
+ report_cfg["example_customers"] = args.example_customers
212
 
213
  if args.mode == "train":
214
  run_train(grpo_config, report_cfg, paths_cfg, args.hf_token)