diff --git a/optuna_dashboard/ts/components/StudyDetail.tsx b/optuna_dashboard/ts/components/StudyDetail.tsx index 2f56ec8b..d928daa4 100644 --- a/optuna_dashboard/ts/components/StudyDetail.tsx +++ b/optuna_dashboard/ts/components/StudyDetail.tsx @@ -150,32 +150,6 @@ export const StudyDetail: FC<{ } else if (page === "trialTable") { content = ( - - - - - - Download CSV File - - - - diff --git a/optuna_dashboard/ts/components/TrialTable.tsx b/optuna_dashboard/ts/components/TrialTable.tsx index efc79626..98a0b176 100644 --- a/optuna_dashboard/ts/components/TrialTable.tsx +++ b/optuna_dashboard/ts/components/TrialTable.tsx @@ -1,6 +1,7 @@ import React, { FC } from "react" -import { IconButton } from "@mui/material" +import { IconButton, Button, useTheme } from "@mui/material" import LinkIcon from "@mui/icons-material/Link" +import DownloadIcon from "@mui/icons-material/Download" import { DataGridColumn, DataGrid } from "./DataGrid" import { Link } from "react-router-dom" @@ -9,6 +10,7 @@ export const TrialTable: FC<{ studyDetail: StudyDetail | null initialRowsPerPage?: number }> = ({ studyDetail, initialRowsPerPage }) => { + const theme = useTheme() const trials: Trial[] = studyDetail !== null ? studyDetail.trials : [] const objectiveNames: string[] = studyDetail?.objective_names || [] @@ -190,12 +192,23 @@ export const TrialTable: FC<{ }) return ( - - columns={columns} - rows={trials} - keyField={"trial_id"} - dense={true} - initialRowsPerPage={initialRowsPerPage} - /> + <> + + columns={columns} + rows={trials} + keyField={"trial_id"} + dense={true} + initialRowsPerPage={initialRowsPerPage} + /> + + ) }