From d9dc79b769094ffed23776b98f1f6bc757644dff Mon Sep 17 00:00:00 2001
From: Kaustubh Maske Patil <37668193+nikochiko@users.noreply.github.com>
Date: Tue, 27 Aug 2024 19:56:06 +0530
Subject: [PATCH] fix: set current_run = new run when we save on submit

---
 daras_ai_v2/base.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
index 424cdff79..9b831f5a2 100644
--- a/daras_ai_v2/base.py
+++ b/daras_ai_v2/base.py
@@ -632,6 +632,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,