fix: set current_run = new run when we save on submit

GooeyAI · Aug 29, 2024 · a77a267 · a77a267
1 parent 8c9aa77
commit a77a267
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
@@ -633,6 +633,11 @@ def _render_publish_modal(
                 gui.error(str(e))
                 return
 
+        if self._has_current_run_changed(current_run):
+            sr = self._on_submit()
+            if sr:
+                current_run = sr
+
         if is_update_mode:
             updates = dict(
                 saved_run=current_run,