From 3eee5f6c417e69a2b96fce0d31d2e330e11c8468 Mon Sep 17 00:00:00 2001
From: Marco Salazar <marco@elementl.com>
Date: Fri, 20 Dec 2024 10:55:22 -0500
Subject: [PATCH] 1 more ting

---
 .../dagster-ui/packages/ui-core/src/runs/Run.tsx | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/js_modules/dagster-ui/packages/ui-core/src/runs/Run.tsx b/js_modules/dagster-ui/packages/ui-core/src/runs/Run.tsx
index fde9c0f26bd65..4fd391633cf5c 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/runs/Run.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/runs/Run.tsx
@@ -105,7 +105,7 @@ export const Run = memo((props: RunProps) => {
       <LogsProvider key={runId} runId={runId}>
         {(logs) => (
           <>
-            <OnLogsLoaded dependency={logsDependency} />
+            <OnLogsLoaded dependency={logsDependency} logs={logs} />
             <RunMetadataProvider logs={logs}>
               {(metadata) => (
                 <RunWithData
@@ -128,10 +128,18 @@ export const Run = memo((props: RunProps) => {
   );
 });
 
-const OnLogsLoaded = ({dependency}: {dependency: ReturnType<typeof useTraceDependency>}) => {
+const OnLogsLoaded = ({
+  dependency,
+  logs,
+}: {
+  dependency: ReturnType<typeof useTraceDependency>;
+  logs: LogsProviderLogs;
+}) => {
   useLayoutEffect(() => {
-    dependency.completeDependency(CompletionType.SUCCESS);
-  }, [dependency]);
+    if (!logs.loading) {
+      dependency.completeDependency(CompletionType.SUCCESS);
+    }
+  }, [dependency, logs]);
   return null;
 };