From bdb6f44ee85235b3191a3225e5a53650a441306c Mon Sep 17 00:00:00 2001
From: Jack Quinn <jack@jq.codes>
Date: Sun, 15 Sep 2024 17:42:47 -0400
Subject: [PATCH] Make loading state local to EnhancedRunDetails

Signed-off-by: Jack Quinn <jack@jq.codes>
---
 frontend/src/pages/RunDetails.tsx       | 5 +++++
 frontend/src/pages/RunDetailsRouter.tsx | 6 +-----
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/frontend/src/pages/RunDetails.tsx b/frontend/src/pages/RunDetails.tsx
index afca9e5f0a4..9d25c137fcb 100644
--- a/frontend/src/pages/RunDetails.tsx
+++ b/frontend/src/pages/RunDetails.tsx
@@ -101,6 +101,7 @@ interface SelectedNodeDetails {
 
 // exported only for testing
 export interface RunDetailsInternalProps {
+  isLoading?: boolean;
   runId?: string;
   gkeMetadata: GkeMetadata;
 }
@@ -237,6 +238,10 @@ class RunDetails extends Page<RunDetailsInternalProps, RunDetailsState> {
   }
 
   public render(): JSX.Element {
+    if (this.props.isLoading) {
+      return <div>Currently loading run information</div>;
+    }
+
     const {
       allArtifactConfigs,
       allowCustomVisualizations,
diff --git a/frontend/src/pages/RunDetailsRouter.tsx b/frontend/src/pages/RunDetailsRouter.tsx
index f9505233c8e..ee6e0a97047 100644
--- a/frontend/src/pages/RunDetailsRouter.tsx
+++ b/frontend/src/pages/RunDetailsRouter.tsx
@@ -70,9 +70,5 @@ export default function RunDetailsRouter(props: RunDetailsProps) {
     }
   }
 
-  if (runIsFetching || templateStrIsFetching) {
-    return <div>Currently loading run information</div>;
-  }
-
-  return <EnhancedRunDetails {...props} />;
+  return <EnhancedRunDetails {...props} isLoading={runIsFetching || templateStrIsFetching} />;
 }