From d0a3832c4931e81aa23d7f8b9b98a4fcd5917e6e Mon Sep 17 00:00:00 2001
From: keisuke-umezawa <keisuke.umezawa@gmail.com>
Date: Thu, 18 Apr 2024 09:34:01 +0900
Subject: [PATCH] Use objective names

---
 optuna_dashboard/ts/components/TrialTable.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/optuna_dashboard/ts/components/TrialTable.tsx b/optuna_dashboard/ts/components/TrialTable.tsx
index e15e0ea8..e7f8f80c 100644
--- a/optuna_dashboard/ts/components/TrialTable.tsx
+++ b/optuna_dashboard/ts/components/TrialTable.tsx
@@ -33,7 +33,7 @@ export const TrialTable: FC<{
 }> = ({ studyDetail }) => {
   const theme = useTheme()
   const trials: Trial[] = studyDetail !== null ? studyDetail.trials : []
-  // TODO: const objectiveNames: string[] = studyDetail?.objective_names || []
+  const objectiveNames: string[] = studyDetail?.objective_names || []
 
   const columnHelper = createColumnHelper<Trial>()
   const columns: ColumnDef<Trial>[] = [
@@ -65,7 +65,10 @@ export const TrialTable: FC<{
       ...studyDetail.directions.map((s, objectiveId) =>
         columnHelper.accessor((row) => row["values"]?.[objectiveId], {
           id: `values_${objectiveId}`,
-          header: `Objective ${objectiveId}`,
+          header:
+            objectiveNames.length === studyDetail?.directions.length
+              ? objectiveNames[objectiveId]
+              : `Objective ${objectiveId}`,
           footer: (info) => info.column.id,
           enableSorting: true,
           enableColumnFilter: false,