Spaces:
Running
Running
fix: Update train_agent and evaluate_agent calls to use new config-based signatures
Browse files- 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 |
-
|
| 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:
|