Merge pull request #162 from optuna/distribution-v3

Add FloatDistribution and IntDitribution for Optuna v3 release.

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 },
     },
   ],

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],

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<{

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"