Spaces:
Running on T4
Running on T4
Claude commited on
Add --llm-agent and other legacy CLI flags for backwards compatibility
Browse filesExternal 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
- 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)
|