RoyAalekh commited on
Commit
e206120
·
1 Parent(s): 5bbed6a

fix: Update train_agent and evaluate_agent calls to use new config-based signatures

Browse files
Files changed (1) hide show
  1. court_scheduler_rl.py +2 -3
court_scheduler_rl.py CHANGED
@@ -213,9 +213,7 @@ class InteractivePipeline:
213
  rl_cfg = self.config.rl_training
214
  training_stats = train_agent(
215
  agent=agent,
216
- episodes=rl_cfg.episodes,
217
- cases_per_episode=rl_cfg.cases_per_episode,
218
- episode_length=rl_cfg.episode_length_days,
219
  verbose=False # Disable internal printing
220
  )
221
 
@@ -244,6 +242,7 @@ class InteractivePipeline:
244
  test_cases=eval_cases,
245
  episodes=5,
246
  episode_length=rl_cfg.episode_length_days,
 
247
  )
248
  self.output.save_evaluation_stats(evaluation_stats)
249
  except Exception as eval_err:
 
213
  rl_cfg = self.config.rl_training
214
  training_stats = train_agent(
215
  agent=agent,
216
+ rl_config=rl_cfg,
 
 
217
  verbose=False # Disable internal printing
218
  )
219
 
 
242
  test_cases=eval_cases,
243
  episodes=5,
244
  episode_length=rl_cfg.episode_length_days,
245
+ rl_config=rl_cfg,
246
  )
247
  self.output.save_evaluation_stats(evaluation_stats)
248
  except Exception as eval_err: