Poll runs after page loaded (#5)

pytorch · Apr 14, 2021 · 131a7e2 · 131a7e2
1 parent ca08488
commit 131a7e2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/tb_plugin/fe/src/app.tsx b/tb_plugin/fe/src/app.tsx
@@ -139,10 +139,20 @@ export const App = () => {
     })
   }, [])
 
+  const waitForRuns = async () => {
+    while (true) {
+      try {
+        const runs = await api.defaultApi.runsGet()
+        setRuns(runs)
+        break
+      } catch (e) {
+        console.info('Cannot fetch runs: ', e)
+      }
+    }
+  }
+
   React.useEffect(() => {
-    api.defaultApi.runsGet().then((runs) => {
-      setRuns(runs)
-    })
+    waitForRuns()
   }, [])
 
   React.useEffect(() => {

diff --git a/tb_plugin/torch_tb_profiler/static/index.html b/tb_plugin/torch_tb_profiler/static/index.html