Skip to content

Commit 5885822

Browse files
committed
fix rft command
1 parent 76e9cec commit 5885822

1 file changed

Lines changed: 8 additions & 6 deletions

File tree

eval_protocol/cli_commands/create_rft.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -681,13 +681,13 @@ def _create_rft_job(
681681
return 1
682682

683683
wandb_config: Optional[Dict[str, Any]] = None
684-
if getattr(args, "wandb_enabled", False):
684+
if getattr(args, "enabled", False):
685685
wandb_config = {
686686
"enabled": True,
687-
"apiKey": getattr(args, "wandb_api_key", None),
688-
"project": getattr(args, "wandb_project", None),
689-
"entity": getattr(args, "wandb_entity", None),
690-
"runId": getattr(args, "wandb_run_id", None),
687+
"apiKey": getattr(args, "api_key", None),
688+
"project": getattr(args, "project", None),
689+
"entity": getattr(args, "entity", None),
690+
"runId": getattr(args, "run_id", None),
691691
}
692692

693693
body: Dict[str, Any] = {
@@ -702,7 +702,9 @@ def _create_rft_job(
702702
"outputStats": None,
703703
"outputMetrics": None,
704704
"mcpServer": getattr(args, "mcp_server", None),
705-
"jobId": getattr(args, "job_id", None),
705+
"jobId": getattr(args, "reinforcement_fine_tuning_job_id", None),
706+
"sourceJob": getattr(args, "source_job", None),
707+
"quiet": getattr(args, "quiet", False),
706708
}
707709
# Debug: print minimal summary
708710
print(f"Prepared RFT job for evaluator '{evaluator_id}' using dataset '{dataset_id}'")

0 commit comments

Comments
 (0)