fix: restore eval_enabled early signal for subprocess training

This commit is contained in:
Roland Tannous 2026-03-09 15:35:49 +00:00
commit 2a11e79b8b
2 changed files with 11 additions and 0 deletions

View file

@ -384,6 +384,9 @@ class TrainingBackend:
self.eval_step_history.append(step)
self.eval_enabled = True
elif etype == "eval_configured":
self.eval_enabled = True
elif etype == "status":
self._progress.status_message = event.get("message", "")
self._progress.is_training = True

View file

@ -269,6 +269,14 @@ def run_training_process(
if eval_steps is not None and float(eval_steps) <= 0:
eval_dataset = None
# Tell the parent process that eval is configured so the frontend
# shows "Waiting for first evaluation step..." instead of "not configured"
if eval_dataset is not None:
event_queue.put({
"type": "eval_configured",
"ts": time.time(),
})
if dataset is None or trainer.should_stop:
if trainer.should_stop:
event_queue.put({"type": "complete", "output_dir": None, "ts": time.time()})