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/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"