added Unknown distribution and convertDistribution in suggestion client

added unit tests Signed-off-by: Shashank Mittal <[email protected]>
kubeflow · Aug 15, 2024 · 3a3540e · 3a3540e
1 parent 51b246f
commit 3a3540e
Show file tree

Hide file tree

Showing 7 changed files with 261 additions and 96 deletions.
diff --git a/pkg/apis/controller/experiments/v1beta1/experiment_types.go b/pkg/apis/controller/experiments/v1beta1/experiment_types.go
@@ -221,6 +221,7 @@ const (
 	DistributionLogUniform Distribution = "logUniform"
 	DistributionNormal     Distribution = "normal"
 	DistributionLogNormal  Distribution = "logNormal"
+	DistributionUnknown    Distribution = "unknown"
 )
 
 // TrialTemplate describes structure of trial template

diff --git a/pkg/apis/manager/v1beta1/api.pb.go b/pkg/apis/manager/v1beta1/api.pb.go
diff --git a/pkg/apis/manager/v1beta1/api.proto b/pkg/apis/manager/v1beta1/api.proto
@@ -105,6 +105,7 @@ enum Distribution {
     LOG_UNIFORM = 1;
     NORMAL = 2;
     LOG_NORMAL = 3;
+    DISTRIBUTION_UNKNOWN = 4;
 }
 
 /**