Merge pull request optuna#791 from knshnb/plot-fetch-n-trials

Fetch plot API when num_trials is updated
keisuke-umezawa · Feb 5, 2024 · c571e1f · c571e1f
2 parents 8f95d59 + 35baab0
commit c571e1f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/optuna_dashboard/ts/components/GraphContour.tsx b/optuna_dashboard/ts/components/GraphContour.tsx
@@ -34,6 +34,8 @@ const ContourBackend: FC<{
   study: StudyDetail | null
 }> = ({ study = null }) => {
   const studyId = study?.id
+  const numCompletedTrials =
+    study?.trials.filter((t) => t.state === "Complete").length || 0
   useEffect(() => {
     if (studyId === undefined) {
       return
@@ -45,7 +47,7 @@ const ContourBackend: FC<{
       .catch((err) => {
         console.error(err)
       })
-  }, [studyId])
+  }, [studyId, numCompletedTrials])
   return <Box id={plotDomId} sx={{ height: "450px" }} />
 }
 

diff --git a/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx b/optuna_dashboard/ts/components/GraphParallelCoordinate.tsx
@@ -100,6 +100,8 @@ const GraphParallelCoordinateBackend: FC<{
   study: StudyDetail | null
 }> = ({ study = null }) => {
   const studyId = study?.id
+  const numCompletedTrials =
+    study?.trials.filter((t) => t.state === "Complete").length || 0
   useEffect(() => {
     if (studyId === undefined) {
       return
@@ -111,7 +113,7 @@ const GraphParallelCoordinateBackend: FC<{
       .catch((err) => {
         console.error(err)
       })
-  }, [studyId])
+  }, [studyId, numCompletedTrials])
   return <Box id={plotDomId} sx={{ height: "450px" }} />
 }
 

diff --git a/optuna_dashboard/ts/components/GraphRank.tsx b/optuna_dashboard/ts/components/GraphRank.tsx
@@ -45,6 +45,8 @@ const GraphRankBackend: FC<{
   study: StudyDetail | null
 }> = ({ study = null }) => {
   const studyId = study?.id
+  const numCompletedTrials =
+    study?.trials.filter((t) => t.state === "Complete").length || 0
   useEffect(() => {
     if (studyId === undefined) {
       return
@@ -56,7 +58,7 @@ const GraphRankBackend: FC<{
       .catch((err) => {
         console.error(err)
       })
-  }, [studyId])
+  }, [studyId, numCompletedTrials])
   return <Box id={plotDomId} sx={{ height: "450px" }} />
 }
 

diff --git a/optuna_dashboard/ts/components/GraphSlice.tsx b/optuna_dashboard/ts/components/GraphSlice.tsx
@@ -46,6 +46,8 @@ const GraphSliceBackend: FC<{
   study: StudyDetail | null
 }> = ({ study = null }) => {
   const studyId = study?.id
+  const numCompletedTrials =
+    study?.trials.filter((t) => t.state === "Complete").length || 0
   useEffect(() => {
     if (studyId === undefined) {
       return
@@ -57,7 +59,7 @@ const GraphSliceBackend: FC<{
       .catch((err) => {
         console.error(err)
       })
-  }, [studyId])
+  }, [studyId, numCompletedTrials])
   return <Box id={plotDomId} sx={{ height: "450px" }} />
 }