diff --git a/frontend_tests/TrialTable.test.tsx b/frontend_tests/TrialTable.test.tsx index f6a9a7d9..4ddd6fc3 100644 --- a/frontend_tests/TrialTable.test.tsx +++ b/frontend_tests/TrialTable.test.tsx @@ -52,24 +52,24 @@ const study_detail = { intersection_search_space: [ { name: "x", - distribution: "UniformDistribution" as Distribution, + distribution: "FloatDistribution" as Distribution, attributes: { low: -3, high: 3 }, }, { name: "y", - distribution: "UniformDistribution" as Distribution, + distribution: "FloatDistribution" as Distribution, attributes: { low: -3, high: 3 }, }, ], union_search_space: [ { name: "x", - distribution: "UniformDistribution" as Distribution, + distribution: "FloatDistribution" as Distribution, attributes: { low: -3, high: 3 }, }, { name: "y", - distribution: "UniformDistribution" as Distribution, + distribution: "FloatDistribution" as Distribution, attributes: { low: -3, high: 3 }, }, ], diff --git a/optuna_dashboard/static/components/GraphHyperparameterImportances.tsx b/optuna_dashboard/static/components/GraphHyperparameterImportances.tsx index 4595effb..888a5eda 100644 --- a/optuna_dashboard/static/components/GraphHyperparameterImportances.tsx +++ b/optuna_dashboard/static/components/GraphHyperparameterImportances.tsx @@ -39,9 +39,11 @@ const plotlyColorsSequentialBlues = [ ] const distributionColors = { + FloatDistribution: plotlyColorsSequentialBlues.slice(-1)[0], UniformDistribution: plotlyColorsSequentialBlues.slice(-1)[0], LogUniformDistribution: plotlyColorsSequentialBlues.slice(-1)[0], DiscreteUniformDistribution: plotlyColorsSequentialBlues.slice(-1)[0], + IntDistribution: plotlyColorsSequentialBlues.slice(-2)[0], IntUniformDistribution: plotlyColorsSequentialBlues.slice(-2)[0], IntLogUniformDistribution: plotlyColorsSequentialBlues.slice(-2)[0], CategoricalDistribution: plotlyColorsSequentialBlues.slice(-4)[0], diff --git a/optuna_dashboard/static/components/GraphSlice.tsx b/optuna_dashboard/static/components/GraphSlice.tsx index cb4bd055..91534c7b 100644 --- a/optuna_dashboard/static/components/GraphSlice.tsx +++ b/optuna_dashboard/static/components/GraphSlice.tsx @@ -26,6 +26,7 @@ const useStyles = makeStyles((theme: Theme) => }) ) +// TODO(c-bata): Check `log` field of IntDistribution and FloatDistribution. const logDistributions = ["LogUniformDistribution", "IntLogUniformDistribution"] export const GraphSlice: FC<{ diff --git a/optuna_dashboard/static/types/index.d.ts b/optuna_dashboard/static/types/index.d.ts index 6eb7279c..614b78f0 100644 --- a/optuna_dashboard/static/types/index.d.ts +++ b/optuna_dashboard/static/types/index.d.ts @@ -10,6 +10,8 @@ declare const URL_PREFIX: string type TrialState = "Running" | "Complete" | "Pruned" | "Fail" | "Waiting" type StudyDirection = "maximize" | "minimize" | "not_set" type Distribution = + | "FloatDistribution" + | "IntDistribution" | "UniformDistribution" | "LogUniformDistribution" | "DiscreteUniformDistribution"