From e85ed2a13a89c167e98de27cff8bf02c9ff143d3 Mon Sep 17 00:00:00 2001
From: c-bata <contact@c-bata.link>
Date: Tue, 1 Feb 2022 22:51:37 +0900
Subject: [PATCH] Add IntDistribution & FloatDistribution

---
 .../static/components/GraphHyperparameterImportances.tsx        | 2 ++
 optuna_dashboard/static/types/index.d.ts                        | 2 ++
 2 files changed, 4 insertions(+)

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"