Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4538 from CalamitousFelicitousness/state-cleanup-patch

Browse Source 
fix(control): ensure state cleanup runs reliably
This commit is contained in:
Vladimir Mandic
2026-01-12 16:12:27 +01:00
committed by GitHub
parent e4e863fd6d ca3aecef2d
commit 1cdb46c455
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ui_control.py
+5 -2
View File
@@ -103,12 +103,15 @@ def generate_click(job_id: str, state: str, active_tab: str, *args):
for results in control_run(state, units, helpers.input_source, helpers.input_init, helpers.input_mask, active_tab, True, *args):
progress.record_results(job_id, results)
yield return_controls(results, t)
except GeneratorExit:
pass # Generator was closed by client - cleanup will run in finally
except Exception as e:
shared.log.error(f"Control exception: {e}")
errors.display(e, 'Control')
yield [None, None, None, None, f'Control: Exception: {e}', '']
progress.finish_task(job_id)
shared.state.end(jobid)
finally:
progress.finish_task(job_id)
shared.state.end(jobid)
def create_ui(_blocks: gr.Blocks=None):