@@ -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