From fe538ac2fd89a676a92dc33b1f1bbabbe804de1e Mon Sep 17 00:00:00 2001
From: Kenshin Abe <abe.kenshin@gmail.com>
Date: Fri, 2 Feb 2024 19:32:53 +0900
Subject: [PATCH] Use numCompletedTrials

---
 optuna_dashboard/ts/components/GraphEdf.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/optuna_dashboard/ts/components/GraphEdf.tsx b/optuna_dashboard/ts/components/GraphEdf.tsx
index 6bb65bdc..dffa7274 100644
--- a/optuna_dashboard/ts/components/GraphEdf.tsx
+++ b/optuna_dashboard/ts/components/GraphEdf.tsx
@@ -29,6 +29,11 @@ const GraphEdfBackend: FC<{
 }> = ({ studies }) => {
   const studyIds = studies.map((s) => s.id)
   const domId = getPlotDomId(-1)
+  const numCompletedTrials = studies.reduce(
+    (acc, study) =>
+      acc + study?.trials.filter((t) => t.state === "Complete").length,
+    0
+  )
   useEffect(() => {
     if (studyIds.length === 0) {
       return
@@ -40,7 +45,7 @@ const GraphEdfBackend: FC<{
       .catch((err) => {
         console.error(err)
       })
-  }, [studyIds])
+  }, [studyIds, numCompletedTrials])
   return <Box id={domId} sx={{ height: "450px" }} />
 }