set default distribution as uniform

Signed-off-by: Shashank Mittal <shashank.mittal.mec22@itbhu.ac.in>
kubeflow · Jan 7, 2025 · c35789b · c35789b
1 parent 4f35663
commit c35789b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pkg/apis/controller/experiments/v1beta1/experiment_defaults.go b/pkg/apis/controller/experiments/v1beta1/experiment_defaults.go
@@ -30,6 +30,7 @@ func (e *Experiment) SetDefault() {
 	e.setDefaultObjective()
 	e.setDefaultTrialTemplate()
 	e.setDefaultMetricsCollector()
+	e.setDefaultParameterDistribution()
 }
 
 func (e *Experiment) setDefaultParallelTrialCount() {
@@ -176,3 +177,11 @@ func (e *Experiment) setDefaultMetricsCollector() {
 		}
 	}
 }
+
+func (e *Experiment) setDefaultParameterDistribution() {
+	for i := range e.Spec.Parameters {
+		if e.Spec.Parameters[i].FeasibleSpace.Distribution == "" {
+			e.Spec.Parameters[i].FeasibleSpace.Distribution = DistributionUniform
+		}
+	}
+}