Merge pull request #19 from xmarre/codex/fix-flux-vae-memory-leak

Fix Spectrum run teardown to release forecaster state
2026-04-01 13:54:23 +02:00
parent 585f7fd398 71e7d9d6ac
commit be5269a1aa
1 changed files with 1 additions and 1 deletions
@@ -203,7 +203,7 @@ class SpectrumRuntime:
        if self._active_run is None or self._active_run.run_id != int(run_id):
            return
        self._active_run = None
-        self._active_steps = {}
+        self._reset_scheduler_state()

    def begin_solver_step(self, run_id: int, solver_step_id: int, time_coord: float, total_steps: int) -> Dict[str, Any]:
        if self._active_run is None or self._active_run.run_id != int(run_id):